Searched refs:pad_shape (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | mirror_pad_op.cc | 68 const TensorShape pad_shape = ctx->InputShape("paddings"); in Compile() local
80 TensorShapeUtils::IsMatrix(pad_shape) && pad_shape.dim_size(1) == 2, in Compile()
82 pad_shape.DebugString())); in Compile()
84 ctx, dims == pad_shape.dim_size(0), in Compile()
87 pad_shape.DebugString(), " ", input_shape.DebugString())); in Compile()
167 const TensorShape pad_shape = ctx->InputShape("paddings"); in Compile() local
179 TensorShapeUtils::IsMatrix(pad_shape) && pad_shape.dim_size(1) == 2, in Compile()
181 pad_shape.DebugString())); in Compile()
183 ctx, dims == pad_shape.dim_size(0), in Compile()
186 pad_shape.DebugString(), " ", input_shape.DebugString())); in Compile()
|
| D | pad_op.cc | 34 const TensorShape pad_shape = ctx->InputShape("paddings"); in Compile() local
38 TensorShapeUtils::IsMatrix(pad_shape) && pad_shape.dim_size(1) == 2, in Compile()
40 pad_shape.DebugString())); in Compile()
42 ctx, dims == pad_shape.dim_size(0), in Compile()
45 pad_shape.DebugString(), " ", input_shape.DebugString())); in Compile()
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_batch_gather_with_default_op.py | 113 pad_shape = _get_pad_shape(params, indices, row_splits_dtype)
116 default_value, pad_shape)
119 pad_shape = array_ops.concat([
125 pad = array_ops.broadcast_to(default_value, pad_shape)
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_space_to_depth_pass.cc | 227 std::vector<int64_t> pad_shape = {filter_shape[0] + pad_h, in GetPadOpForConv2DFilter() local
230 SmallVector<int64_t, 4> expand_shape(pad_shape.begin(), pad_shape.end()); in GetPadOpForConv2DFilter()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | convolution_handler.cc | 357 Shape pad_shape = lhs.hlo()->shape(); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local
434 pad_shape.set_dimensions(lhs_dimension, in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
456 HloInstruction::CreatePad(pad_shape, lhs.hlo(), zero, pad_config)); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
|
| D | spmd_partitioner.cc | 2597 const Shape& pad_shape = in HandleInfeed() local
2599 if (ShapeUtil::Compatible(element_shape, pad_shape)) { in HandleInfeed()
2603 CHECK(pad_shape.IsArray()); in HandleInfeed()
2604 return PadToShape(infeed_element, pad_shape, &branch_b); in HandleInfeed()
2608 return CreateZero(pad_shape, &branch_b); in HandleInfeed()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_creation_utils.cc | 74 Shape pad_shape, in MakePadHlo()
78 pad_shape, operand, padding_value, padding_config)); in MakePadHlo()
|
| D | algebraic_simplifier.cc | 3420 auto pad_shape = pad->shape(); in HandlePad() local
3422 auto pad_shape1 = pad_shape; in HandlePad()
3448 HloInstruction::CreateBroadcast(pad_shape, pad2, padding_dimensions)); in HandlePad()
|
| D | algebraic_simplifier_test.cc | 5092 const Shape pad_shape, in TEST_P()
5096 pad_shape, input, in TEST_P()
|