| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | sparse_tensor.h | 329 static inline int GetSliceIndex(const int dim, const int split_size, in GetSliceIndex() argument
331 DCHECK_GT(split_size, 0); in GetSliceIndex()
333 if (residual == 0) return dim / split_size; in GetSliceIndex()
334 const int offset = residual * (split_size + 1); in GetSliceIndex()
336 return dim / (split_size + 1); in GetSliceIndex()
338 return residual + ((dim - offset) / split_size); in GetSliceIndex()
343 static inline int GetDimensionInSlice(const int dim, const int split_size, in GetDimensionInSlice() argument
345 DCHECK_GT(split_size, 0); in GetDimensionInSlice()
347 if (residual == 0) return dim % split_size; in GetDimensionInSlice()
348 const int offset = residual * (split_size + 1); in GetDimensionInSlice()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | concat_lib_gpu_impl.cu.cc | 38 GpuDeviceArrayStruct<const T*> input_ptr_data, int split_size, in concat_fixed_kernel() argument
46 IntType split = gidx / split_size; in concat_fixed_kernel()
48 IntType col_offset = gidx % split_size; in concat_fixed_kernel()
52 input_ptr[gidy * split_size + col_offset]; in concat_fixed_kernel()
141 bool fixed_size, int split_size, in ConcatGPUImpl() argument
149 gpu_device.stream()>>>(input_ptrs, split_size, output->dimension(0), in ConcatGPUImpl()
193 int split_size, typename TTypes<T, 2>::Matrix* output);
200 int split_size, typename TTypes<T, 2>::Matrix* output);
|
| D | split_op.cc | 240 auto make_sizes = [&](Eigen::DenseIndex split_size) { in Compute() argument
241 return Eigen::DSizes<Eigen::DenseIndex, 2>{split_size, suffix_dim_size}; in Compute()
243 auto reshape_result = [&](Tensor* result, Eigen::DenseIndex split_size) { in Compute() argument
244 return result->shaped<T, 2>({split_size, suffix_dim_size}); in Compute()
253 auto make_sizes = [&](Eigen::DenseIndex split_size) { in Compute() argument
254 return Eigen::DSizes<Eigen::DenseIndex, 3>{prefix_dim_size, split_size, in Compute()
257 auto reshape_result = [&](Tensor* result, Eigen::DenseIndex split_size) { in Compute() argument
259 {prefix_dim_size, split_size, suffix_dim_size}); in Compute()
|
| D | split_v_op.cc | 302 auto make_sizes = [&](Eigen::DenseIndex split_size) { in Compute() argument
303 return Eigen::DSizes<Eigen::DenseIndex, 2>{split_size, suffix_dim_size}; in Compute()
305 auto reshape_result = [&](Tensor* result, Tlen split_size) { in Compute() argument
306 return result->shaped<T, 2>({split_size, suffix_dim_size}); in Compute()
315 auto make_sizes = [&](Eigen::DenseIndex split_size) { in Compute() argument
316 return Eigen::DSizes<Eigen::DenseIndex, 3>{prefix_dim_size, split_size, in Compute()
319 auto reshape_result = [&](Tensor* result, Tlen split_size) { in Compute() argument
321 {prefix_dim_size, split_size, suffix_dim_size}); in Compute()
|
| D | concat_lib_gpu.h | 62 int split_size, typename TTypes<T, 2>::Matrix* output); \
67 int split_size, typename TTypes<T, 2>::Matrix* output);
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_array_ops.py | 217 split_size = math_ops.cumprod(mask_shape) + 1
220 masked_splits = math_ops.range(split_size[dim]) * elt_size
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_ops.py | 518 split_size = total_grad_size // num_splits
519 split_size_last = total_grad_size - split_size * (num_splits - 1)
520 split_sizes = [split_size] * (num_splits - 1) + [split_size_last]
|
| /external/libaom/libaom/av1/decoder/ |
| D | decodeframe.c | 1244 int split_size, int txs, int blk_row, in set_inter_tx_size() argument
1246 for (int idy = 0; idy < tx_size_high_unit[split_size]; in set_inter_tx_size()
1248 for (int idx = 0; idx < tx_size_wide_unit[split_size]; in set_inter_tx_size()
|