Home
last modified time | relevance | path

Searched defs:padded_shape (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_constant_slice.cc66 Shape padded_shape = input_array.shape(); in Slice() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dspacetobatch_op.cc54 std::vector<int64> padded_shape(input_shape.begin(), input_shape.end()); in SpaceToBatch() local
/external/tensorflow/tensorflow/core/kernels/data/
Dpadded_batch_dataset_op.cc246 const PartialTensorShape& padded_shape = in GetNextInternal() local
428 PartialTensorShape padded_shape; in MakeDataset() local
Dpadded_batch_dataset_op_test.cc54 for (auto& padded_shape : padded_shapes_) { in GetInputTensors() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dtree_reduction_rewriter.cc131 Shape padded_shape = in RewriteReduction() local
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_fused_image_transform.cc709 TensorShape padded_shape; in Compute() local
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.cc250 HloInstruction* PadToShape(HloInstruction* hlo, const Shape& padded_shape, in PadToShape()
613 auto padded_shape = result->shape(); in PadFromPartialReplicateShape() local
Ddot_handler.cc799 auto padded_shape = padded_slice_operand->shape(); in PartitionBaseCase() local
3062 Shape padded_shape = user_gte->shape(); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local
Dspmd_partitioner.cc578 auto padded_shape = base_shape_; in ReshardAsWindowedInput() local
1142 auto padded_shape = hlo_->shape(); in ReshardWithAllToAll() local
/external/tensorflow/tensorflow/python/data/ops/
Ddataset_ops.py4083 def _is_padded_shape_compatible_with(padded_shape, input_component_shape): argument
4107 def _padded_shape_to_tensor(padded_shape, input_component_shape): argument
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlower_tf.cc876 auto padded_shape = llvm::to_vector<4>(input_shape); in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_padder.cc752 Shape padded_shape = input->shape(); in RewriteInputWithDynamicPadding() local
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_common.cc843 SmallVector<int64_t, 2> padded_shape(input_rank); in convertSpaceToBatchNDOp() local