Searched refs:vshape (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| H A D | cpu_fusion_test.cc | 54 Shape vshape = input_literal1.shape(); in TEST_F() local
62 HloInstruction::CreateBinary(vshape, HloOpcode::kAdd, input1, input2)); in TEST_F()
64 HloInstruction::CreateUnary(vshape, HloOpcode::kNegate, add1)); in TEST_F()
93 Shape vshape = input_literal.shape(); in TEST_F() local
98 HloInstruction::CreateUnary(vshape, HloOpcode::kNegate, input)); in TEST_F()
100 HloInstruction::CreateUnary(vshape, HloOpcode::kCeil, negate)); in TEST_F()
102 HloInstruction::CreateUnary(vshape, HloOpcode::kExp, ceil)); in TEST_F()
104 HloInstruction::CreateUnary(vshape, HloOpcode::kFloor, exp)); in TEST_F()
106 vshape, in TEST_F()
111 HloInstruction::CreateBinary(vshape, HloOpcode::kMultiply, two, floor)); in TEST_F()
[all …]
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/python/ |
| H A D | python_ir.cpp | 830 auto vshape = t.expectRef<TensorType>().sizes(); in initPythonIRBindings() local
831 return vshape.size() ? py::cast(*vshape.size()) in initPythonIRBindings()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/variables/ |
| H A D | resource_variable_ops_test.py | 201 vshape = resource_variable_ops.variable_shape(v.handle)
203 tensor_util.constant_value(vshape),
206 self.assertEqual("Const", vshape.op.type)
|