Searched defs:dynamic_shape (Results 1 – 10 of 10) 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/xla/service/ |
| D | executable.cc | 45 Status ExecutionInput::SetDynamicShape(Shape dynamic_shape) { in SetDynamicShape()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_execute.cc | 167 bool DynamicShapeIsCompatible(const xla::Shape& dynamic_shape, in DynamicShapeIsCompatible()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | hlo_legalize_to_lhlo.cc | 351 SmallVector<int64_t, 2> dynamic_shape(result_rank, in InsertDynamicMemrefCastOp() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | strided_slice_op.cc | 565 xla::XlaOp dynamic_shape = ctx->Input(0); in Compile() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.cc | 1576 const xla::Shape& dynamic_shape, const xla::Shape& bounded_shape) { in DynamicArrayShapeIsCompatible()
1589 const xla::Shape& dynamic_shape, const xla::Shape& bounded_shape) { in DynamicShapeIsCompatible()
|
| D | literal_test.cc | 758 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 | 259 XLA_Shape dynamic_shape; member
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 1367 llvm::SmallVector<int64_t, 8> dynamic_shape(shape_ty.getDimSize(0), in GetReshapeOutputType() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 595 llvm::SmallVector<int64_t, 8> dynamic_shape(shape_ty.getDimSize(0), in GetReshapeOutputType() local
|