Searched defs:dynamic_shape (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_util.cc | 93 for (auto& dynamic_shape : dynamic_shapes) { in DynamicShapesToTensorShapes() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | shape_simplification.cc | 192 ShapeOfOp dynamic_shape; in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_resource_ops_test.cc | 59 bool dynamic_shape; member
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | executable.cc | 45 Status ExecutionInput::SetDynamicShape(Shape dynamic_shape) { in SetDynamicShape()
|
| D | shape_inference_test.cc | 2372 auto dynamic_shape = ShapeUtil::MakeShape(S32, {4}, {true}); in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | hlo_legalize_to_memref.cc | 260 SmallVector<int64_t, 2> dynamic_shape(result_rank, in InsertDynamicMemrefCastOp() local
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_execute.cc | 167 bool DynamicShapeIsCompatible(const xla::Shape& dynamic_shape, in DynamicShapeIsCompatible()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | py_buffer.cc | 153 Shape dynamic_shape; in xla_dynamic_shape() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | strided_slice_op.cc | 603 xla::XlaOp dynamic_shape = ctx->Input(0); in Compile() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.cc | 1616 const xla::Shape& dynamic_shape, const xla::Shape& bounded_shape) { in DynamicArrayShapeIsCompatible()
1629 const xla::Shape& dynamic_shape, const xla::Shape& bounded_shape) { in DynamicShapeIsCompatible()
|
| D | literal_test.cc | 767 auto dynamic_shape = ShapeUtil::MakeShape(F32, {2, 3}, {false, true}); in TEST_F() local
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | c_api_decl.h | 262 XLA_Shape dynamic_shape; member
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 726 llvm::SmallVector<int64_t, 8> dynamic_shape(shape_ty.getDimSize(0), in GetReshapeOutputType() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 1628 llvm::SmallVector<int64_t, 8> dynamic_shape(shape_ty.getDimSize(0), in GetReshapeOutputType() local
|