Home
last modified time | relevance | path

Searched defs:scalar_shape (Results 1 – 25 of 32) 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
418 const Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_F() local
466 const Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_F() local
Dcollective_decomposer_utils.cc45 const Shape &scalar_shape = zero->shape(); in CreateStartIndicesForCollectiveDecomposition() local
Dsort_simplifier.cc90 Shape scalar_shape = in RemoveUnusedOperandFromSort() local
Dstable_sort_expander.cc84 Shape scalar_shape = ShapeUtil::MakeShape(S32, {}); in ExpandInstruction() local
Dhlo_creation_utils.cc389 auto scalar_shape = ShapeUtil::MakeShape(operand->shape().element_type(), {}); in MakeReduceHlo() local
403 auto scalar_shape = ShapeUtil::MakeShape(operand->shape().element_type(), {}); in MakeReduceHlo() local
429 auto scalar_shape = ShapeUtil::MakeShape(operand->shape().element_type(), {}); in MakeReduceHlo() local
Dhlo_live_range_test.cc266 Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_F() local
Dhlo_verifier_test.cc76 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
95 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
114 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
Dhlo_evaluator_test.cc160 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in MaxComputationScalarF32() local
2566 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
2601 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in BM_ReducePrecisely() local
2643 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_P() local
2802 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_P() local
2855 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_P() local
Dpattern_matcher_test.cc71 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
888 const Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_F() local
Dhlo_cse_test.cc593 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
Dbuffer_assignment_test.cc1391 auto scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
1461 auto scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
1763 auto scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
Dalgebraic_simplifier_test.cc622 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
749 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
1838 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
1881 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
2669 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
5744 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_P() local
Dmemory_space_assignment_test.cc776 Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_P() local
2585 Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_P() local
2908 Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_P() local
3000 Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_P() 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/dtensor/mlir/
Dvalue_utils.cc78 mlir::Value scalar_shape = in ReshapeSizeTypeToScalar() local
216 mlir::Value scalar_shape = mlir::TF::collection_ops_util::GetR1Const( in SelectScalarValueFromArray() local
/external/tensorflow/tensorflow/cc/experimental/libtf/tests/
Dvariable_test.cc90 PartialTensorShape scalar_shape; in TEST_P() local
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dcomparators.cc72 auto scalar_shape = ShapeUtil::MakeShape(operand_type, {}); in CreateScalarComparisonComputation() local
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dmatmul_builder.cc52 static int scalar_shape[] = {1, 1, 1, 1}; in AddFullyConnectedHelper() local
/external/tensorflow/tensorflow/c/
Dops_test.cc324 shape_inference::ShapeHandle* scalar_shape = in TEST() local
/external/tensorflow/tensorflow/core/kernels/
Dfake_quant_ops.cc275 TensorShape scalar_shape; in Compute() local
/external/tensorflow/tensorflow/dtensor/mlir/expansions/
Dsave_restore_spmd_expander.cc161 mlir::Value scalar_shape = mlir::TF::collection_ops_util::GetR1Const( in DeviceIdToLocalBranchIndex() local
Dmeta_spmd_expander.cc1057 mlir::Value scalar_shape = mlir::TF::collection_ops_util::GetR1Const( in ExpandOp() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dmap_test.cc316 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in XLA_TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/client/
Dvalue_inference.cc1303 Shape scalar_shape = ShapeUtil::MakeScalarShape(xla::PRED); in AnalyzeIsDynamic() local

12