/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_ordering_test.cc | 56 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
|
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 | 71 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
|
D | hlo_live_range_test.cc | 267 Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_F() local
|
D | hlo_evaluator_test.cc | 134 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
|
D | pattern_matcher_test.cc | 72 auto scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
|
D | heap_simulator_test.cc | 43 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
|
D | hlo_cse_test.cc | 615 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
|
D | hlo_creation_utils.cc | 305 auto scalar_shape = ShapeUtil::MakeShape(operand->shape().element_type(), {}); in MakeReduceHlo() local
|
D | memory_space_assignment_test.cc | 551 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
|
D | algebraic_simplifier_test.cc | 438 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
|
D | buffer_assignment_test.cc | 1379 auto scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local 1734 auto scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
|
D | hlo_instruction_test.cc | 754 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
|
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
D | xla_reduce_op.cc | 74 xla::Shape scalar_shape; in Compile() local
|
D | reduce_window_op.cc | 91 xla::Shape scalar_shape; in Compile() local
|
D | xla_select_and_scatter_op.cc | 98 xla::Shape scalar_shape; in Compile() local
|
/external/tensorflow/tensorflow/lite/experimental/delegates/hexagon/builders/ |
D | neg_op_builder.cc | 25 static int scalar_shape[] = {1, 1, 1, 1}; in PopulateSubGraph() local
|
D | activation_builder.cc | 31 static int scalar_shape[] = {1, 1, 1, 1}; in PopulateSubGraph() 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 | 112 auto scalar_shape = ShapeUtil::MakeShape(operand_type, {}); in CreateScalarComparisonComputation() local
|
/external/tensorflow/tensorflow/core/kernels/ |
D | fake_quant_ops.cc | 253 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/core/grappler/optimizers/ |
D | auto_mixed_precision_test.cc | 882 Output scalar_shape = ops::Const(s.WithOpName("scalar_shape"), 0, {0}); in TEST_F() local
|
/external/tensorflow/tensorflow/compiler/xrt/tests/ |
D | raw_api_test.cc | 1593 const xla::Shape scalar_shape = xla::ShapeUtil::MakeShape(xla::F32, {}); in TEST() local
|