Home
last modified time | relevance | path

Searched defs:scalar_shape (Results 1 – 25 of 28) 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
129 const Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_F() local
179 const Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_F() local
212 const Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_F() local
417 const Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_F() local
465 const Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_F() local
Dsort_simplifier.cc90 Shape scalar_shape = in RemoveUnusedOperandFromSort() local
Dstable_sort_expander.cc84 Shape scalar_shape = ShapeUtil::MakeShape(S32, {}); in ExpandInstruction() local
Dhlo_verifier_test.cc71 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
90 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
109 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
Dhlo_creation_utils.cc326 auto scalar_shape = ShapeUtil::MakeShape(operand->shape().element_type(), {}); in MakeReduceHlo() local
355 auto scalar_shape = ShapeUtil::MakeShape(operand->shape().element_type(), {}); in MakeReduceHlo() local
Dhlo_live_range_test.cc266 Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_F() local
Dpattern_matcher_test.cc72 auto scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
Dhlo_evaluator_test.cc135 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in MaxComputationScalarF32() local
2528 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
2563 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in BM_ReducePrecisely() local
2605 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_P() local
2764 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_P() local
2817 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_P() local
Dhlo_cse_test.cc615 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
Dheap_simulator_test.cc44 const Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_F() local
888 const Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_F() local
Dalgebraic_simplifier_test.cc593 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
720 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
1768 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
1811 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
2254 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
4362 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
4448 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
5104 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_P() local
Dmemory_space_assignment_test.cc741 Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_P() local
2624 Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_P() local
2947 Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_P() local
3039 Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_P() local
Dbuffer_assignment_test.cc1372 auto scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
1727 auto scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
Dhlo_instruction_test.cc812 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dreduce_window_op.cc91 xla::Shape scalar_shape; in Compile() local
Dxla_select_and_scatter_op.cc98 xla::Shape scalar_shape; in Compile() local
/external/tensorflow/tensorflow/core/framework/
Dvariant_op_copy_test.cc177 TensorShape scalar_shape({}); in TEST() local
211 TensorShape scalar_shape({}); in TEST() local
254 TensorShape scalar_shape({}); in TEST() local
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dcomparators.cc72 auto scalar_shape = ShapeUtil::MakeShape(operand_type, {}); in CreateScalarComparisonComputation() local
/external/tensorflow/tensorflow/c/
Dops_test.cc324 shape_inference::ShapeHandle* scalar_shape = in TEST() local
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dmatmul_builder.cc52 static int scalar_shape[] = {1, 1, 1, 1}; in AddFullyConnectedHelper() local
/external/tensorflow/tensorflow/core/kernels/
Dfake_quant_ops.cc253 TensorShape scalar_shape; in Compute() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dmap_test.cc315 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in XLA_TEST_F() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dauto_mixed_precision_test.cc895 Output scalar_shape = ops::Const(s.WithOpName("scalar_shape"), 0, {0}); in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.cc684 auto scalar_shape = ShapeUtil::MakeShape(S32, {}); in Calculate() local
/external/tensorflow/tensorflow/compiler/xrt/tests/
Draw_api_test.cc2100 const xla::Shape scalar_shape = xla::ShapeUtil::MakeShape(xla::F32, {}); in TEST() local

12