Home
last modified time | relevance | path

Searched defs:scalar_shape (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_ordering_test.cc57 const Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_F() local
130 const Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_F() local
180 const Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_F() local
213 const Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_F() local
411 const Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_F() local
459 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.cc70 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
89 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
108 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
Dhlo_evaluator_test.cc1442 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
1478 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in BM_ReducePrecisely() local
1519 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_P() local
1561 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_P() local
1613 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_P() local
1664 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_P() local
1717 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_P() local
Dheap_simulator_test.cc43 const Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_F() local
695 const Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_F() local
Dhlo_creation_utils.cc258 auto scalar_shape = ShapeUtil::MakeShape(operand->shape().element_type(), {}); in MakeReduceHlo() local
Dpattern_matcher_test.cc69 auto scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST() local
Dhlo_cse_test.cc615 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
Dalgebraic_simplifier_test.cc372 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
478 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
1404 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
1447 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
1727 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
3595 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
3681 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
4247 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_P() local
Dbuffer_assignment_test.cc1200 auto scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
1555 auto scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
Dhlo_instruction_test.cc752 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
Dshape_inference_test.cc932 const Shape scalar_shape = ShapeUtil::MakeShape(element_type, {}); in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_hlo_support_checker_test.cc38 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_hlo_support_checker_test.cc38 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dxla_reduce_op.cc75 xla::Shape scalar_shape; in Compile() local
Dreduce_window_op.cc92 xla::Shape scalar_shape; in Compile() local
Dxla_select_and_scatter_op.cc99 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.cc113 auto scalar_shape = ShapeUtil::MakeShape(operand_type, {}); in CreateScalarComparisonComputation() local
/external/tensorflow/tensorflow/python/framework/
Dcommon_shapes.py43 def scalar_shape(unused_op): function
/external/tensorflow/tensorflow/core/kernels/
Dfake_quant_ops.cc241 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/compiler/xla/client/
Dxla_builder.cc1647 auto scalar_shape = ShapeUtil::MakeShape(operand_type, {}); in Sort() local
2124 const Shape& scalar_shape = ShapeUtil::MakeShape(shape.element_type(), {}); in CrossReplicaSum() local