Home
last modified time | relevance | path

Searched defs:scalar_shape (Results 1 – 25 of 27) 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_live_range_test.cc267 Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_F() local
Dhlo_evaluator_test.cc134 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in MaxComputationScalarF32() local
2527 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
2563 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in BM_ReducePrecisely() local
2604 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_P() local
2763 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_P() local
2816 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_P() local
Dpattern_matcher_test.cc72 auto scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
Dheap_simulator_test.cc43 const Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_F() local
883 const Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_F() local
Dhlo_cse_test.cc615 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
Dhlo_creation_utils.cc305 auto scalar_shape = ShapeUtil::MakeShape(operand->shape().element_type(), {}); in MakeReduceHlo() local
Dmemory_space_assignment_test.cc551 Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_P() local
1502 Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_P() local
1824 Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_P() local
1916 Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_P() local
Dalgebraic_simplifier_test.cc438 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
565 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
1630 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
1673 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
2042 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
3983 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
4069 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
4657 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_P() local
Dbuffer_assignment_test.cc1379 auto scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
1734 auto scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
Dhlo_instruction_test.cc754 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dxla_reduce_op.cc74 xla::Shape scalar_shape; in Compile() local
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/lite/experimental/delegates/hexagon/builders/
Dneg_op_builder.cc25 static int scalar_shape[] = {1, 1, 1, 1}; in PopulateSubGraph() local
Dactivation_builder.cc31 static int scalar_shape[] = {1, 1, 1, 1}; in PopulateSubGraph() 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.cc112 auto scalar_shape = ShapeUtil::MakeShape(operand_type, {}); in CreateScalarComparisonComputation() 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.cc882 Output scalar_shape = ops::Const(s.WithOpName("scalar_shape"), 0, {0}); in TEST_F() local
/external/tensorflow/tensorflow/compiler/xrt/tests/
Draw_api_test.cc1593 const xla::Shape scalar_shape = xla::ShapeUtil::MakeShape(xla::F32, {}); in TEST() local

12