Searched defs:padded_shape (Results 1 – 13 of 13) sorted by relevance
66 Shape padded_shape = input_array.shape(); in Slice() local
54 std::vector<int64> padded_shape(input_shape.begin(), input_shape.end()); in SpaceToBatch() local
246 const PartialTensorShape& padded_shape = in GetNextInternal() local428 PartialTensorShape padded_shape; in MakeDataset() local
54 for (auto& padded_shape : padded_shapes_) { in GetInputTensors() local
131 Shape padded_shape = in RewriteReduction() local
709 TensorShape padded_shape; in Compute() local
250 HloInstruction* PadToShape(HloInstruction* hlo, const Shape& padded_shape, in PadToShape()613 auto padded_shape = result->shape(); in PadFromPartialReplicateShape() local
799 auto padded_shape = padded_slice_operand->shape(); in PartitionBaseCase() local3062 Shape padded_shape = user_gte->shape(); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local
578 auto padded_shape = base_shape_; in ReshardAsWindowedInput() local1142 auto padded_shape = hlo_->shape(); in ReshardWithAllToAll() local
4083 def _is_padded_shape_compatible_with(padded_shape, input_component_shape): argument4107 def _padded_shape_to_tensor(padded_shape, input_component_shape): argument
876 auto padded_shape = llvm::to_vector<4>(input_shape); in matchAndRewrite() local
752 Shape padded_shape = input->shape(); in RewriteInputWithDynamicPadding() local
843 SmallVector<int64_t, 2> padded_shape(input_rank); in convertSpaceToBatchNDOp() local