Searched defs:dynamic_shape (Results 1 – 13 of 13) 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 | 61 bool dynamic_shape; member
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | executable.cc | 44 Status ExecutionInput::SetDynamicShape(Shape dynamic_shape) { in SetDynamicShape()
|
| D | shape_inference_test.cc | 2386 auto dynamic_shape = ShapeUtil::MakeShape(S32, {4}, {true}); in TEST_F() local
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_execute.cc | 169 bool DynamicShapeIsCompatible(const xla::Shape& dynamic_shape, in DynamicShapeIsCompatible()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | strided_slice_op.cc | 604 xla::XlaOp dynamic_shape = ctx->Input(0); in Compile() local
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | py_buffer.cc | 158 Shape dynamic_shape; in xla_dynamic_shape() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.cc | 1596 const xla::Shape& dynamic_shape, const xla::Shape& bounded_shape) { in DynamicArrayShapeIsCompatible()
1609 const xla::Shape& dynamic_shape, const xla::Shape& bounded_shape) { in DynamicShapeIsCompatible()
|
| D | literal_test.cc | 765 auto dynamic_shape = ShapeUtil::MakeShape(F32, {2, 3}, {false, true}); in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/ |
| D | c_api_decl.h | 286 XLA_Shape dynamic_shape; member
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 765 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 | 1773 llvm::SmallVector<int64_t, 8> dynamic_shape(shape_ty.getDimSize(0), in GetReshapeOutputType() local
|