Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A 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
H A 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
H A Dcollective_decomposer_utils.cc45 const Shape &scalar_shape = zero->shape(); in CreateStartIndicesForCollectiveDecomposition() local
H A Dsort_simplifier.cc90 Shape scalar_shape = in RemoveUnusedOperandFromSort() local
H A Dstable_sort_expander.cc84 Shape scalar_shape = ShapeUtil::MakeShape(S32, {}); in ExpandInstruction() local
H A 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
H A 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
H A Dhlo_live_range_test.cc266 Shape scalar_shape = ShapeUtil::MakeShape(xla::F32, {}); in TEST_F() local
H A 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
H A 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
H A Dpattern_matcher_test.cc71 auto scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
H A Dhlo_cse_test.cc593 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
H A 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
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dvalue_utils.cc78 mlir::Value scalar_shape = in ReshapeSizeTypeToScalar() local
216 mlir::Value scalar_shape = mlir::TF::collection_ops_util::GetR1Const( in SelectScalarValueFromArray() local
/aosp_15_r20/external/tensorflow/tensorflow/cc/experimental/libtf/tests/
H A Dvariable_test.cc90 PartialTensorShape scalar_shape; in TEST_P() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/
H A Dcomparators.cc72 auto scalar_shape = ShapeUtil::MakeShape(operand_type, {}); in CreateScalarComparisonComputation() local
/aosp_15_r20/external/tensorflow/tensorflow/c/
H A Dops_test.cc324 shape_inference::ShapeHandle* scalar_shape = in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
H A Dmatmul_builder.cc52 static int scalar_shape[] = {1, 1, 1, 1}; in AddFullyConnectedHelper() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dfake_quant_ops.cc275 TensorShape scalar_shape; in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/
H A Dsave_restore_spmd_expander.cc161 mlir::Value scalar_shape = mlir::TF::collection_ops_util::GetR1Const( in DeviceIdToLocalBranchIndex() local
H A Dmeta_spmd_expander.cc1057 mlir::Value scalar_shape = mlir::TF::collection_ops_util::GetR1Const( in ExpandOp() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dmap_test.cc316 Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in XLA_TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dvalue_inference.cc1303 Shape scalar_shape = ShapeUtil::MakeScalarShape(xla::PRED); in AnalyzeIsDynamic() local

12