| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_scheduling_test.cc | 37 const Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); local
39 ShapeUtil::MakeTupleShape({scalar_shape, scalar_shape});
46 HloInstruction::CreateGetTupleElement(scalar_shape, cond_param, 0));
48 HloInstruction::CreateGetTupleElement(scalar_shape, cond_param, 1));
66 HloInstruction::CreateParameter(0, scalar_shape, "param_iter"));
68 HloInstruction::CreateParameter(1, scalar_shape, "param_data"));
|
| hlo_verifier_test.cc | 40 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); local
42 HloInstruction::CreateParameter(0, scalar_shape, "param"));
44 HloInstruction::CreateUnary(scalar_shape, HloOpcode::kNegate, param));
59 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); local
61 HloInstruction::CreateParameter(0, scalar_shape, "param"));
63 HloInstruction::CreateUnary(scalar_shape, HloOpcode::kNegate, param));
78 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); local
80 HloInstruction::CreateParameter(0, scalar_shape, "param"));
82 HloInstruction::CreateUnary(scalar_shape, HloOpcode::kNegate, param));
88 HloInstruction::CreateParameter(0, scalar_shape, "param"))
[all...] |
| hlo_ordering_test.cc | 103 const Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); local
113 HloInstruction::CreateParameter(0, scalar_shape, "param"));
115 HloInstruction::CreateCall(scalar_shape, {}, computation_c));
121 HloInstruction::CreateCall(scalar_shape, {}, computation_c));
127 HloInstruction::CreateCall(scalar_shape, {}, computation_a));
129 HloInstruction::CreateCall(scalar_shape, {x}, computation_b));
176 const Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); local
180 HloInstruction::CreateParameter(0, scalar_shape, "body_param"));
182 scalar_shape, HloOpcode::kNegate, body_param));
187 HloInstruction::CreateParameter(0, scalar_shape, "cond_param"))
239 const Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); local
[all...] |
| hlo_evaluator_test.cc | 1227 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); local
1269 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); local
1320 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); local
1373 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); local
[all...] |
| hlo_cse_test.cc | 427 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); local
434 scalar_shape, RandomDistribution::RNG_UNIFORM, {constant1, constant2}));
438 scalar_shape, HloOpcode::kAdd, rng, param));
|
| heap_simulator_test.cc | 416 const Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); local
418 ShapeUtil::MakeTupleShape({scalar_shape, scalar_shape});
424 HloInstruction::CreateGetTupleElement(scalar_shape, cond_param, 0));
426 HloInstruction::CreateGetTupleElement(scalar_shape, cond_param, 1));
[all...] |
| algebraic_simplifier_test.cc | 941 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); local
2252 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); local
2505 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); local
[all...] |
| hlo_instruction_test.cc | 748 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); local
754 HloInstruction::CreateParameter(0, scalar_shape, "param"));
765 scalar_shape, {constant}, computation_x, /*static_operands=*/{}));
767 scalar_shape, {map_1_x}, computation_x, /*static_operands=*/{}));
769 scalar_shape, {map_2_x}, computation_y, /*static_operands=*/{}));
[all...] |
| buffer_assignment_test.cc | 1037 auto scalar_shape = ShapeUtil::MakeShape(F32, {}); local
1385 auto scalar_shape = ShapeUtil::MakeShape(F32, {}); local
[all...] |
| shape_inference_test.cc | 869 const Shape scalar_shape = ShapeUtil::MakeShape(element_type, {}); local
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_hlo_support_checker_test.cc | 38 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); local
40 HloInstruction::CreateParameter(0, scalar_shape, "param0"));
42 HloInstruction::CreateParameter(1, scalar_shape, "param1"));
44 scalar_shape, HloOpcode::kAdd, param0, param1));
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| gpu_hlo_support_checker_test.cc | 38 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); local
40 HloInstruction::CreateParameter(0, scalar_shape, "param0"));
42 HloInstruction::CreateParameter(1, scalar_shape, "param1"));
44 scalar_shape, HloOpcode::kAdd, param0, param1));
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| list_ops_test.py | 38 def scalar_shape(): function
46 element_shape=scalar_shape())
59 element_shape=scalar_shape())
73 l = list_ops.tensor_list_from_tensor(t, element_shape=scalar_shape())
88 l = list_ops.tensor_list_from_tensor(t, element_shape=scalar_shape())
113 l = list_ops.tensor_list_from_tensor(t, element_shape=scalar_shape())
160 element_shape=scalar_shape())
172 l = list_ops.tensor_list_from_tensor(c, element_shape=scalar_shape())
182 l = list_ops.tensor_list_from_tensor(c, element_shape=scalar_shape())
195 l = list_ops.tensor_list_from_tensor(c, element_shape=scalar_shape())
[all...] |
| /external/tensorflow/tensorflow/core/framework/ |
| variant_op_copy_test.cc | 176 TensorShape scalar_shape({});
177 scalar_shape.AsProto(variant_proto.mutable_tensor_shape());
210 TensorShape scalar_shape({});
211 scalar_shape.AsProto(variant_proto.mutable_tensor_shape());
253 TensorShape scalar_shape({});
254 scalar_shape.AsProto(variant_proto.mutable_tensor_shape());
|
| /external/tensorflow/tensorflow/core/kernels/ |
| fake_quant_ops.cc | 241 TensorShape scalar_shape; variable
244 context->allocate_output(1, scalar_shape, &grad_wrt_min));
248 context->allocate_output(2, scalar_shape, &grad_wrt_max));
|
| /external/tensorflow/tensorflow/python/eager/ |
| ops_test.py | 281 scalar_shape = constant_op.constant([], dtype=dtypes.int32)
283 x = random_ops.random_uniform(scalar_shape)
288 scalar_shape, minval=constant_op.constant(5.),
|
| /external/tensorflow/tensorflow/contrib/tensorrt/convert/ |
| convert_nodes.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| map_test.cc | 314 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); local
321 auto embed4_param = embed4_builder.Parameter(0, scalar_shape, "x");
|
| /external/tensorflow/tensorflow/python/framework/ |
| common_shapes.py | 31 def scalar_shape(unused_op): function
|
| ops_test.py | [all...] |
