Searched refs:pad_start (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | spacetobatch_functor.cc | 41 const int64* pad_start, const int64* block_offsets, in run()
47 batch_tensor_pos * block_shape[0] + block_offsets[0] - pad_start[0]; in run()
52 pad_start + 1, block_offsets + 1, batch_tensor_shape + 1, in run()
72 const int64* pad_start, const int64* block_offsets, in run()
103 int64 pad_start[NUM_BLOCK_DIMS]; in operator ()() local
108 pad_start[block_dim] = paddings_tensor[block_dim * 2]; in operator ()()
145 space_tensor_shape, &space_tensor_strides[1], block_shape, pad_start, in operator ()()
|
| D | spacetobatch_functor_gpu.cu.cc | 37 int32 pad_start[NUM_BLOCK_DIMS]; member
74 offset - args.pad_start[block_dim]; in S2B()
131 args.pad_start[block_dim] = paddings[block_dim * 2]; in operator ()()
|
| D | spacetobatch_op.cc | 148 const int64 pad_start = paddings[2 * block_dim], in SpaceToBatchOpCompute() local
150 if (pad_start < 0 || pad_end < 0) { in SpaceToBatchOpCompute()
155 const int64 padded_size = input_size + pad_start + pad_end; in SpaceToBatchOpCompute()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | spacetobatch_op.cc | 59 int64 pad_start = paddings.Get<int64>({i, 0}); in SpaceToBatch() local
61 OP_REQUIRES(ctx, pad_start >= 0 && pad_end >= 0, in SpaceToBatch()
63 dim->set_edge_padding_low(pad_start); in SpaceToBatch()
65 padded_shape[1 + i] += pad_start + pad_end; in SpaceToBatch()
|
| /external/webrtc/rtc_base/third_party/base64/ |
| D | base64.cc | 143 size_t byte_len = 0, pad_len = 0, pad_start = 0; in GetNextQuantum() local
165 pad_start = *dpos; in GetNextQuantum()
187 *dpos = pad_start; in GetNextQuantum()
|
| /external/pigweed/pw_presubmit/py/pw_presubmit/ |
| D | tools.py | 78 pad_start: str = ' ',
100 width += len(pad_start)
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_SpaceToBatchND.pbtxt | 20 `paddings[i] = [pad_start, pad_end]` specifies the padding for input dimension
22 `block_shape[i]` divides `input_shape[i + 1] + pad_start + pad_end`.
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 1259 const int64_t pad_start = in Verify() local
1263 if (pad_start < 0 || pad_end < 0) { in Verify()
1276 const int64_t pad_start = in Verify() local
1282 if ((input_len + pad_start + pad_end) % block_len != 0) { in Verify()
|
| D | tf_generated_ops.td | 13981 `paddings[i] = [pad_start, pad_end]` specifies the padding for input dimension
13983 `block_shape[i]` divides `input_shape[i + 1] + pad_start + pad_end`.
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 3935 pad_start = base_paddings[:, 0]
3937 full_input_shape = input_shape + pad_start + orig_pad_end
3942 [[pad_start[i], pad_end[i]] for i in range(num_block_dims)],
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 2281 const int64 pad_start = paddings_vec[dim * 2], in SpaceToBatchShapeHelper() local
2283 if (pad_start < 0 || pad_end < 0) { in SpaceToBatchShapeHelper()
2289 c->Add(c->Dim(input_shape, dim + 1), pad_start, &padded_size)); in SpaceToBatchShapeHelper()
|