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.cc52 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.cc72 TensorShape arg_shape; in GetTensorShapeFromXlaArgument() local
506 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.cc1228 Shape arg_shape = ShapeUtil::MakeShape(F32, {5, 3}); in TEST_F() local
1239 Shape arg_shape = ShapeUtil::MakeShape(F32, {5, 3}); in TEST_F() local
1250 Shape arg_shape = ShapeUtil::MakeShape(F32, {5, 3}); in TEST_F() local
1261 Shape arg_shape = ShapeUtil::MakeShape(F32, {5, 3}); in TEST_F() local
1628 Shape arg_shape = ShapeUtil::MakeShape(F32, {5, 3}); in TEST_F() local
1639 Shape arg_shape = ShapeUtil::MakeShape(F32, {5, 3}); in TEST_F() local
Dshape_inference.cc391 const Shape* arg_shape = nullptr; in InferConcatOpShape() local
1124 const Shape* arg_shape = arg_shapes[0]; in InferMapShape() local
3197 const Shape& arg_shape = *arg_shapes[i]; in InferCallShape() local
Dhlo_evaluator.cc247 const auto& arg_shape = arg_literals[i]->shape(); in Evaluate() local
2366 const Shape& arg_shape = input_args[0]->shape(); in GenerateReduceOutputElement() local
2435 const Shape& arg_shape = input_args[0]->shape(); in HandleReduce() local
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc1768 InferredShape& arg_shape = in GetArgAndRetvalShapes() local
2364 const InferredShape& arg_shape = arg_shapes[i]; in BuildCompileNode() local
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc1450 xla::Shape& arg_shape = arg_shapes->back(); in SetEntryTupleShapesAndLeafReplication() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler_test.cc1621 xla::Shape arg_shape = xla::ShapeUtil::MakeTupleShape(shapes); in TEST_F() local