Home
last modified time | relevance | path

Searched refs:scalar_shape (Results 1 – 25 of 31) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_ordering_test.cc56 const Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_F() local
66 HloInstruction::CreateParameter(0, scalar_shape, "param")); in TEST_F()
68 HloInstruction::CreateCall(scalar_shape, {}, computation_c)); in TEST_F()
74 HloInstruction::CreateCall(scalar_shape, {}, computation_c)); in TEST_F()
80 HloInstruction::CreateCall(scalar_shape, {}, computation_a)); in TEST_F()
82 HloInstruction::CreateCall(scalar_shape, {x}, computation_b)); in TEST_F()
129 const Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_F() local
133 HloInstruction::CreateParameter(0, scalar_shape, "body_param")); in TEST_F()
135 scalar_shape, HloOpcode::kNegate, body_param)); in TEST_F()
140 HloInstruction::CreateParameter(0, scalar_shape, "cond_param")); in TEST_F()
[all …]
Dhlo_live_range_test.cc266 Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_F() local
267 Shape tuple_shape = ShapeUtil::MakeTupleShape({shape, scalar_shape}); in TEST_F()
273 HloInstruction::CreateGetTupleElement(scalar_shape, cond_param, 1)); in TEST_F()
286 HloInstruction::CreateGetTupleElement(scalar_shape, body_param, 1)); in TEST_F()
293 scalar_shape, HloOpcode::kAdd, body_iter, body_iter_increment)); in TEST_F()
315 HloInstruction::CreateParameter(1, scalar_shape, "param_data")); in TEST_F()
Dsort_simplifier.cc90 Shape scalar_shape = in RemoveUnusedOperandFromSort() local
93 parameter_number, scalar_shape, in RemoveUnusedOperandFromSort()
97 parameter_number, scalar_shape, in RemoveUnusedOperandFromSort()
Dhlo_verifier_test.cc71 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
73 HloInstruction::CreateParameter(0, scalar_shape, "param")); in TEST_F()
75 HloInstruction::CreateUnary(scalar_shape, HloOpcode::kNegate, param)); in TEST_F()
90 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
92 HloInstruction::CreateParameter(0, scalar_shape, "param")); in TEST_F()
94 HloInstruction::CreateUnary(scalar_shape, HloOpcode::kNegate, param)); in TEST_F()
109 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
111 HloInstruction::CreateParameter(0, scalar_shape, "param")); in TEST_F()
113 HloInstruction::CreateUnary(scalar_shape, HloOpcode::kNegate, param)); in TEST_F()
119 HloInstruction::CreateParameter(0, scalar_shape, "param")); in TEST_F()
Dhlo_creation_utils.cc326 auto scalar_shape = ShapeUtil::MakeShape(operand->shape().element_type(), {}); in MakeReduceHlo() local
334 HloInstruction::CreateParameter(0, scalar_shape, "lhs")); in MakeReduceHlo()
336 HloInstruction::CreateParameter(1, scalar_shape, "rhs")); in MakeReduceHlo()
338 HloInstruction::CreateBinary(scalar_shape, binary_opcode, lhs, rhs)); in MakeReduceHlo()
355 auto scalar_shape = ShapeUtil::MakeShape(operand->shape().element_type(), {}); in MakeReduceHlo() local
360 HloInstruction::CreateParameter(0, scalar_shape, "lhs")); in MakeReduceHlo()
362 HloInstruction::CreateParameter(1, scalar_shape, "rhs")); in MakeReduceHlo()
364 HloInstruction::CreateBinary(scalar_shape, binary_opcode, lhs, rhs)); in MakeReduceHlo()
369 scalar_shape, operand, init_value, all_dims, reduce_computation)); in MakeReduceHlo()
Dstable_sort_expander.cc84 Shape scalar_shape = ShapeUtil::MakeShape(S32, {}); in ExpandInstruction() local
86 sort->operand_count() * 2, scalar_shape, in ExpandInstruction()
90 sort->operand_count() * 2 + 1, scalar_shape, in ExpandInstruction()
Dpattern_matcher_test.cc72 auto scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
74 EXPECT_TRUE(Match(&scalar_shape, match::Shape(&matched_shape).IsScalar())); in TEST_F()
75 EXPECT_EQ(matched_shape, &scalar_shape); in TEST_F()
76 EXPECT_TRUE(Match(&scalar_shape, match::Shape().IsArray())); in TEST_F()
77 EXPECT_TRUE(Match(&scalar_shape, match::Shape().IsDenseArray())); in TEST_F()
78 EXPECT_FALSE(Match(&scalar_shape, match::Shape().IsTuple())); in TEST_F()
79 EXPECT_TRUE(Match(&scalar_shape, match::Shape().WithElementType(F32))); in TEST_F()
80 EXPECT_TRUE(Match(&scalar_shape, match::Shape().WithRank(0))); in TEST_F()
82 &scalar_shape, in TEST_F()
Dhlo_evaluator_test.cc135 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in MaxComputationScalarF32() local
137 HloInstruction::CreateParameter(0, scalar_shape, "lhs")); in MaxComputationScalarF32()
139 HloInstruction::CreateParameter(1, scalar_shape, "rhs")); in MaxComputationScalarF32()
141 scalar_shape, HloOpcode::kMaximum, param_lhs, param_rhs)); in MaxComputationScalarF32()
2528 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
2530 HloInstruction::CreateParameter(0, scalar_shape, "lhs")); in TEST_F()
2532 HloInstruction::CreateParameter(1, scalar_shape, "rhs")); in TEST_F()
2534 scalar_shape, HloOpcode::kAdd, param_lhs, param_rhs)); in TEST_F()
2538 HloInstruction::CreateReduce(scalar_shape, arg_instruction, init_value, in TEST_F()
2563 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in BM_ReducePrecisely() local
[all …]
Dheap_simulator_test.cc44 const Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_F() local
46 ShapeUtil::MakeTupleShape({scalar_shape, scalar_shape}); in TEST_F()
53 HloInstruction::CreateGetTupleElement(scalar_shape, cond_param, 0)); in TEST_F()
55 HloInstruction::CreateGetTupleElement(scalar_shape, cond_param, 1)); in TEST_F()
73 HloInstruction::CreateParameter(0, scalar_shape, "param_iter")); in TEST_F()
75 HloInstruction::CreateParameter(1, scalar_shape, "param_data")); in TEST_F()
888 const Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_F() local
890 ShapeUtil::MakeTupleShape({scalar_shape, scalar_shape}); in TEST_F()
896 HloInstruction::CreateGetTupleElement(scalar_shape, cond_param, 0)); in TEST_F()
898 HloInstruction::CreateGetTupleElement(scalar_shape, cond_param, 1)); in TEST_F()
Dmemory_space_assignment_test.cc741 Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_P() local
742 Shape tuple_shape = ShapeUtil::MakeTupleShape({shape, scalar_shape}); in TEST_P()
749 HloInstruction::CreateGetTupleElement(scalar_shape, cond_param, 1)); in TEST_P()
764 HloInstruction::CreateGetTupleElement(scalar_shape, body_param, 1)); in TEST_P()
771 scalar_shape, HloOpcode::kAdd, body_iter, body_iter_increment)); in TEST_P()
793 HloInstruction::CreateParameter(1, scalar_shape, "param_data")); in TEST_P()
821 EXPECT_THAT(iter, op::ShapeWithLayout(scalar_shape)); in TEST_P()
823 EXPECT_THAT(body_iter, op::ShapeWithLayout(scalar_shape)); in TEST_P()
824 EXPECT_THAT(cond_iter, op::ShapeWithLayout(scalar_shape)); in TEST_P()
2624 Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_P() local
[all …]
Dalgebraic_simplifier_test.cc593 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
595 HloInstruction::CreateParameter(0, scalar_shape, "p0")); in TEST_F()
597 HloInstruction::CreateParameter(1, scalar_shape, "p1")); in TEST_F()
599 HloInstruction::CreateBinary(scalar_shape, HloOpcode::kAdd, p0, p1)); in TEST_F()
720 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
722 HloInstruction::CreateParameter(0, scalar_shape, "p0")); in TEST_F()
724 HloInstruction::CreateParameter(1, scalar_shape, "p1")); in TEST_F()
726 HloInstruction::CreateBinary(scalar_shape, HloOpcode::kAdd, p0, p1)); in TEST_F()
1768 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
1770 HloInstruction::CreateParameter(0, scalar_shape, "p0")); in TEST_F()
[all …]
Dhlo_cse_test.cc615 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
622 scalar_shape, RandomDistribution::RNG_UNIFORM, {constant1, constant2})); in TEST_F()
626 scalar_shape, HloOpcode::kAdd, rng, param)); in TEST_F()
Dbuffer_assignment_test.cc1372 auto scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
1377 HloInstruction::CreateParameter(0, scalar_shape, "map_param")); in TEST_F()
1379 HloInstruction::CreateUnary(scalar_shape, HloOpcode::kNegate, map_param)); in TEST_F()
1727 auto scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
1729 HloInstruction::CreateParameter(0, scalar_shape, "param0")); in TEST_F()
1732 HloInstruction::CreateGetTupleElement(scalar_shape, tuple, 0)); in TEST_F()
1734 scalar_shape, HloOpcode::kCopy, tuple_element)); in TEST_F()
Dhlo_instruction_test.cc812 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
818 HloInstruction::CreateParameter(0, scalar_shape, "param")); in TEST_F()
829 HloInstruction::CreateMap(scalar_shape, {constant}, computation_x)); in TEST_F()
831 HloInstruction::CreateMap(scalar_shape, {map_1_x}, computation_x)); in TEST_F()
833 HloInstruction::CreateMap(scalar_shape, {map_2_x}, computation_y)); in TEST_F()
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dmatmul_builder.cc52 static int scalar_shape[] = {1, 1, 1, 1}; in AddFullyConnectedHelper() local
60 scalar_shape, reinterpret_cast<char*>(&data_min), sizeof(data_min)); in AddFullyConnectedHelper()
62 scalar_shape, reinterpret_cast<char*>(&data_max), sizeof(data_max)); in AddFullyConnectedHelper()
83 matmul_op->AddOutput(sizeof(float), 4, scalar_shape); in AddFullyConnectedHelper()
85 matmul_op->AddOutput(sizeof(float), 4, scalar_shape); in AddFullyConnectedHelper()
97 scalar_shape, reinterpret_cast<char*>(&bias_min), sizeof(bias_min)); in AddFullyConnectedHelper()
99 scalar_shape, reinterpret_cast<char*>(&bias_max), sizeof(bias_max)); in AddFullyConnectedHelper()
115 bias_add_op->AddOutput(sizeof(float), 4, scalar_shape); in AddFullyConnectedHelper()
117 bias_add_op->AddOutput(sizeof(float), 4, scalar_shape); in AddFullyConnectedHelper()
125 scalar_shape, reinterpret_cast<char*>(&output_min), sizeof(output_min)); in AddFullyConnectedHelper()
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dreduce_window_op.cc91 xla::Shape scalar_shape; in Compile() local
93 TensorShapeToXLAShape(dtype, TensorShape(), &scalar_shape)); in Compile()
96 xla::ShapeUtil::Compatible(reducer.xla_output_shape, scalar_shape), in Compile()
99 xla::ShapeUtil::HumanString(scalar_shape), " got ", in Compile()
Dxla_select_and_scatter_op.cc98 xla::Shape scalar_shape; in Compile() local
100 TensorShapeToXLAShape(dtype, TensorShape(), &scalar_shape)); in Compile()
103 xla::ShapeUtil::Compatible(scatter.xla_output_shape, scalar_shape), in Compile()
106 xla::ShapeUtil::HumanString(scalar_shape), " got ", in Compile()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dcomparators.cc72 auto scalar_shape = ShapeUtil::MakeShape(operand_type, {}); in CreateScalarComparisonComputation() local
73 auto lhs_param = Parameter(b.get(), parameter_count * 2, scalar_shape, in CreateScalarComparisonComputation()
75 auto rhs_param = Parameter(b.get(), parameter_count * 2 + 1, scalar_shape, in CreateScalarComparisonComputation()
/external/tensorflow/tensorflow/core/framework/
Dvariant_op_copy_test.cc177 TensorShape scalar_shape({}); in TEST() local
178 scalar_shape.AsProto(variant_proto.mutable_tensor_shape()); in TEST()
211 TensorShape scalar_shape({}); in TEST() local
212 scalar_shape.AsProto(variant_proto.mutable_tensor_shape()); in TEST()
254 TensorShape scalar_shape({}); in TEST() local
255 scalar_shape.AsProto(variant_proto.mutable_tensor_shape()); in TEST()
/external/tensorflow/tensorflow/core/kernels/
Dfake_quant_ops.cc253 TensorShape scalar_shape; in Compute() local
256 context->allocate_output(1, scalar_shape, &grad_wrt_min)); in Compute()
260 context->allocate_output(2, scalar_shape, &grad_wrt_max)); in Compute()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dprocess_broadcast_shapes.h45 RuntimeShape scalar_shape(dims_count, 1); in ProcessBroadcastShapes()
/external/tensorflow/tensorflow/c/
Dops_test.cc324 shape_inference::ShapeHandle* scalar_shape = in TEST() local
326 ASSERT_EQ("[]", c.DebugString(*scalar_shape)); in TEST()
/external/tensorflow/tensorflow/python/eager/
Dops_test.py315 scalar_shape = constant_op.constant([], dtype=dtypes.int32)
317 x = random_ops.random_uniform(scalar_shape)
322 scalar_shape, minval=constant_op.constant(5.),
/external/tensorflow/tensorflow/core/kernels/mlir_generated/
Dbase_binary_ops_test.h165 TensorShape scalar_shape{}; in TestOneScalar()
183 RunAndExpectResult<T, OutT>(op_name, scalar_shape, scalar_input_vector, in TestOneScalar()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dmap_test.cc315 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in XLA_TEST_F() local
322 auto embed4_param = Parameter(&embed4_builder, 0, scalar_shape, "x"); in XLA_TEST_F()

12