Home
last modified time | relevance | path

Searched defs:arg_shape (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/python_tests/
H A Dpython_test_attrs.cc88 const auto& arg_shape = arg_type.getShape(); in verifyRegionArgAttribute() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
H A Dcompile_mlir_util.cc88 TensorShape arg_shape; in GetTensorShapeFromXlaArgument() local
681 for (const auto& arg_shape : arg_shapes) in CompileSerializedMlirToXlaHlo() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dshape_inference_test.cc1263 Shape arg_shape = ShapeUtil::MakeShape(F32, {5, 3}); in TEST_F() local
1274 Shape arg_shape = ShapeUtil::MakeShape(F32, {5, 3}); in TEST_F() local
1285 Shape arg_shape = ShapeUtil::MakeShape(F32, {5, 3}); in TEST_F() local
1296 Shape arg_shape = ShapeUtil::MakeShape(F32, {5, 3}); in TEST_F() local
1663 Shape arg_shape = ShapeUtil::MakeShape(F32, {5, 3}); in TEST_F() local
1674 Shape arg_shape = ShapeUtil::MakeShape(F32, {5, 3}); in TEST_F() local
H A Dshape_inference.cc372 const Shape* arg_shape = nullptr; in InferConcatOpShape() local
1122 const Shape* arg_shape = arg_shapes[0]; in InferMapShape() local
3316 const Shape& arg_shape = *arg_shapes[i]; in InferCallShape() local
H A Dhlo_evaluator.cc786 const auto& arg_shape = arg_literals[i]->shape(); in Evaluate() local
3671 const Shape& arg_shape = input_args[0]->shape(); in GenerateReduceOutputElement() local
3740 const Shape& arg_shape = input_args[0]->shape(); in HandleReduce() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dshape_inference.cc83 const InferredShape& arg_shape = it->second; in PropagateShapes() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dspmd_expansion.cc224 llvm::ArrayRef<int64_t> arg_shape = ranked_type.getShape(); in UpdateFunctionArgsUsingLayout() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dclient.cc234 for (const auto& arg_shape : argument_shapes) { in Compile() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Ddistributed_tpu_rewrite_pass.cc1981 InferredShape& arg_shape = in GetArgAndRetvalShapes() local
2648 const InferredShape& arg_shape = arg_shapes[i]; in BuildCompileNode() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dxla_compiler_test.cc1700 xla::Shape arg_shape = xla::ShapeUtil::MakeTupleShape(shapes); in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/
H A Dmlir_hlo_to_hlo.cc2328 xla::Shape& arg_shape = arg_shapes->back(); in SetEntryTupleShapesAndLeafReplication() local