Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlower_general_dot.cc41 llvm::ArrayRef<int64_t> arg_shape, in TransposeReshape()
/external/tensorflow/tensorflow/compiler/jit/
Dshape_inference.cc83 const InferredShape& arg_shape = it->second; in PropagateShapes() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dcompile_mlir_util.cc73 TensorShape arg_shape; in GetTensorShapeFromXlaArgument() local
532 for (const auto& arg_shape : arg_shapes) in CompileSerializedMlirToXlaHlo() local
/external/tensorflow/tensorflow/compiler/xla/client/
Dclient.cc235 for (const auto& arg_shape : argument_shapes) { in Compile() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference_test.cc1249 Shape arg_shape = ShapeUtil::MakeShape(F32, {5, 3}); in TEST_F() local
1260 Shape arg_shape = ShapeUtil::MakeShape(F32, {5, 3}); in TEST_F() local
1271 Shape arg_shape = ShapeUtil::MakeShape(F32, {5, 3}); in TEST_F() local
1282 Shape arg_shape = ShapeUtil::MakeShape(F32, {5, 3}); in TEST_F() local
1649 Shape arg_shape = ShapeUtil::MakeShape(F32, {5, 3}); in TEST_F() local
1660 Shape arg_shape = ShapeUtil::MakeShape(F32, {5, 3}); in TEST_F() local
Dshape_inference.cc380 const Shape* arg_shape = nullptr; in InferConcatOpShape() local
1113 const Shape* arg_shape = arg_shapes[0]; in InferMapShape() local
3334 const Shape& arg_shape = *arg_shapes[i]; in InferCallShape() local
Dhlo_evaluator.cc248 const auto& arg_shape = arg_literals[i]->shape(); in Evaluate() local
2430 const Shape& arg_shape = input_args[0]->shape(); in GenerateReduceOutputElement() local
2499 const Shape& arg_shape = input_args[0]->shape(); in HandleReduce() local
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc1645 xla::Shape& arg_shape = arg_shapes->back(); in SetEntryTupleShapesAndLeafReplication() local
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc2040 InferredShape& arg_shape = in GetArgAndRetvalShapes() local
2701 const InferredShape& arg_shape = arg_shapes[i]; in BuildCompileNode() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler_test.cc1688 xla::Shape arg_shape = xla::ShapeUtil::MakeTupleShape(shapes); in TEST_F() local