| /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 | collective_decomposer_utils.cc | 40 const Shape &scalar_shape = zero->shape(); in CreateStartIndicesForCollectiveDecomposition() 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_creation_utils.cc | 326 auto scalar_shape = ShapeUtil::MakeShape(operand->shape().element_type(), {}); in MakeReduceHlo() local
357 auto scalar_shape = ShapeUtil::MakeShape(operand->shape().element_type(), {}); in MakeReduceHlo() local
|
| D | hlo_live_range_test.cc | 266 Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_F() 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_evaluator_test.cc | 135 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
|
| D | pattern_matcher_test.cc | 72 auto scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
|
| D | hlo_cse_test.cc | 615 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
|
| D | heap_simulator_test.cc | 44 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
|
| D | algebraic_simplifier_test.cc | 594 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
721 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
1769 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
1812 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
2287 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
4395 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
4481 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
5137 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_P() local
|
| D | buffer_assignment_test.cc | 1374 auto scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
1444 auto scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
1766 auto scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
|
| D | memory_space_assignment_test.cc | 760 Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_P() local
2643 Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_P() local
2966 Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_P() local
3058 Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_P() 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/tf2xla/kernels/ |
| 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/cc/experimental/libtf/tests/ |
| D | variable_test.cc | 90 PartialTensorShape scalar_shape; in TEST_P() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | comparators.cc | 72 auto scalar_shape = ShapeUtil::MakeShape(operand_type, {}); in CreateScalarComparisonComputation() local
|
| /external/tensorflow/tensorflow/c/ |
| D | ops_test.cc | 324 shape_inference::ShapeHandle* scalar_shape = in TEST() local
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | matmul_builder.cc | 52 static int scalar_shape[] = {1, 1, 1, 1}; in AddFullyConnectedHelper() 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/compiler/xla/client/ |
| D | value_inference.cc | 1147 Shape scalar_shape = ShapeUtil::MakeScalarShape(xla::PRED); in AnalyzeIsDynamic() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | auto_mixed_precision_test.cc | 899 Output scalar_shape = ops::Const(s.WithOpName("scalar_shape"), 0, {0}); in TEST_F() local
|