Searched refs:pad_start (Results 1 – 8 of 8) 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 | 38 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/webrtc/base/ |
| D | base64.cc | 142 size_t byte_len = 0, pad_len = 0, pad_start = 0; in GetNextQuantum() local
164 pad_start = *dpos; in GetNextQuantum()
186 *dpos = pad_start; in GetNextQuantum()
|
| /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/python/ops/ |
| D | array_ops.py | 2667 pad_start = base_paddings[:, 0]
2669 full_input_shape = input_shape + pad_start + orig_pad_end
2674 [[pad_start[i], pad_end[i]] for i in range(num_block_dims)],
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 2124 const int64 pad_start = paddings_vec[dim * 2], in SpaceToBatchShapeHelper() local
2126 if (pad_start < 0 || pad_end < 0) { in SpaceToBatchShapeHelper()
2132 c->Add(c->Dim(input_shape, dim + 1), pad_start, &padded_size)); in SpaceToBatchShapeHelper()
|