Searched refs:pad_dims (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_batch_gather_with_default_op.py | 158 pad_dims = params_shape.partitioned_dim_sizes[:-1] + (
161 pad_dims, [])
170 pad_dims = None
172 pad_dims = (constant_op.constant(1, dtype=dtypes.int64),) + (
181 pad_dims = batch_dimensions + (
186 pad_dims, params_shape.inner_dim_sizes)
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 2374 std::vector<int> pad_dims; in ConvertStridedSliceHelper() local
2377 pad_dims.push_back(i); in ConvertStridedSliceHelper()
2380 if (pad_dims.empty()) { in ConvertStridedSliceHelper()
2392 } else if (pad_dims.size() == 1) { in ConvertStridedSliceHelper()
2396 if (pad_dims[0] != 2) { in ConvertStridedSliceHelper()
2397 pad_dims.push_back(2); in ConvertStridedSliceHelper()
2399 pad_dims.push_back(3); in ConvertStridedSliceHelper()
2401 } else if (pad_dims.size() > 2) { in ConvertStridedSliceHelper()
2407 std::sort(pad_dims.begin(), pad_dims.end()); in ConvertStridedSliceHelper()
2412 for (int i = 0; i < pad_dims.size(); i++) { in ConvertStridedSliceHelper()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_2d.h | 47 const Eigen::array<Eigen::IndexPair<IndexType>, Dims>& pad_dims, in operator()
50 output.device(d) = input.inflate(strides).pad(pad_dims).shuffle(order); in operator()
|
| D | conv_grad_filter_ops.cc | 921 const Eigen::array<Eigen::IndexPair<int>, 4>& pad_dims, \
|
| D | conv_grad_input_ops.cc | 1064 const Eigen::array<Eigen::IndexPair<int>, 4>& pad_dims, \
|