/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_ordering_test.cc | 57 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
|
D | sort_simplifier.cc | 90 Shape scalar_shape = in RemoveUnusedOperandFromSort() local
|
D | stable_sort_expander.cc | 84 Shape scalar_shape = ShapeUtil::MakeShape(S32, {}); in ExpandInstruction() local
|
D | hlo_verifier_test.cc | 70 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
|
D | hlo_evaluator_test.cc | 1442 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
|
D | heap_simulator_test.cc | 43 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
|
D | hlo_creation_utils.cc | 258 auto scalar_shape = ShapeUtil::MakeShape(operand->shape().element_type(), {}); in MakeReduceHlo() local
|
D | pattern_matcher_test.cc | 69 auto scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST() local
|
D | hlo_cse_test.cc | 615 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
|
D | algebraic_simplifier_test.cc | 372 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
|
D | buffer_assignment_test.cc | 1200 auto scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local 1555 auto scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
|
D | hlo_instruction_test.cc | 752 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
|
D | shape_inference_test.cc | 932 const Shape scalar_shape = ShapeUtil::MakeShape(element_type, {}); in TEST_F() local
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | cpu_hlo_support_checker_test.cc | 38 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | gpu_hlo_support_checker_test.cc | 38 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
|
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
D | xla_reduce_op.cc | 75 xla::Shape scalar_shape; in Compile() local
|
D | reduce_window_op.cc | 92 xla::Shape scalar_shape; in Compile() local
|
D | xla_select_and_scatter_op.cc | 99 xla::Shape scalar_shape; in Compile() local
|
/external/tensorflow/tensorflow/core/framework/ |
D | variant_op_copy_test.cc | 177 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/ |
D | comparators.cc | 113 auto scalar_shape = ShapeUtil::MakeShape(operand_type, {}); in CreateScalarComparisonComputation() local
|
/external/tensorflow/tensorflow/python/framework/ |
D | common_shapes.py | 43 def scalar_shape(unused_op): function
|
/external/tensorflow/tensorflow/core/kernels/ |
D | fake_quant_ops.cc | 241 TensorShape scalar_shape; in Compute() local
|
/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | map_test.cc | 315 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in XLA_TEST_F() local
|
/external/tensorflow/tensorflow/compiler/xla/client/ |
D | xla_builder.cc | 1647 auto scalar_shape = ShapeUtil::MakeShape(operand_type, {}); in Sort() local 2124 const Shape& scalar_shape = ShapeUtil::MakeShape(shape.element_type(), {}); in CrossReplicaSum() local
|