Searched refs:split_dim_size (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | split_op.cc | 119 IndexType split_dim_size = in SetDims() local
126 return std::make_tuple(prefix_dim_size, split_dim_size, suffix_dim_size); in SetDims()
138 Eigen::DenseIndex split_dim_size, in operator ()() argument
229 Eigen::DenseIndex split_dim_size; in Compute() local
232 std::tie(prefix_dim_size, split_dim_size, suffix_dim_size) = in Compute()
235 const int64 split_dim_output_size = split_dim_size / num_split; in Compute()
239 input.shaped<T, 2>({split_dim_size, suffix_dim_size}); in Compute()
248 split_dim_size, suffix_dim_size, make_sizes, reshape_result, in Compute()
252 {prefix_dim_size, split_dim_size, suffix_dim_size}); in Compute()
263 split_dim_size, suffix_dim_size, make_sizes, reshape_result, in Compute()
[all …]
|
| D | split_v_op.cc | 173 IndexType split_dim_size = in SetDims() local
180 return std::make_tuple(prefix_dim_size, split_dim_size, suffix_dim_size); in SetDims()
193 Eigen::DenseIndex split_dim_size, in operator ()() argument
284 Eigen::DenseIndex split_dim_size; in Compute() local
287 std::tie(prefix_dim_size, split_dim_size, suffix_dim_size) = in Compute()
301 input.shaped<T, 2>({split_dim_size, suffix_dim_size}); in Compute()
310 prefix_dim_size, split_dim_size, suffix_dim_size, split_sizes_vec, in Compute()
314 {prefix_dim_size, split_dim_size, suffix_dim_size}); in Compute()
325 prefix_dim_size, split_dim_size, suffix_dim_size, split_sizes_vec, in Compute()
360 int32 split_dim_size; in Compute() local
[all …]
|
| D | split_lib_gpu.cu.cc | 79 int32 split_dim_size, int32 suffix_dim_size, in SplitOpKernel() argument
86 eigen_assert(split_dim_size % num_split == 0); in SplitOpKernel()
88 int32 size = prefix_dim_size * split_dim_size * suffix_dim_size; in SplitOpKernel()
89 int32 piece_size = split_dim_size / num_split; in SplitOpKernel()
93 int32 i = offset / (split_dim_size * suffix_dim_size); in SplitOpKernel()
94 int32 j = (offset % (split_dim_size * suffix_dim_size)) / suffix_dim_size; in SplitOpKernel()
199 int32 prefix_dim_size, int32 split_dim_size, in Run() argument
203 prefix_dim_size * split_dim_size * suffix_dim_size, d); in Run()
207 prefix_dim_size, split_dim_size, suffix_dim_size, in Run()
|
| D | split_lib_gpu.h | 35 int32 split_dim_size, int32 suffix_dim_size,
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | sparse_tensor.h | 591 const int split_dim_size = input_tensor.shape()[split_dim]; in Split() local
592 const int split_size = split_dim_size / num_split; in Split()
594 if (!(num_split > 0 && num_split <= split_dim_size)) { in Split()
597 split_dim_size, "]")); in Split()
605 const int residual = split_dim_size % num_split; in Split()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 583 DimensionHandle split_dim_size; in __anon7c94107b0a02() local
586 true /* evenly_divisible */, &split_dim_size), in __anon7c94107b0a02()
589 c->ReplaceDim(input, split_dim, split_dim_size, &out)); in __anon7c94107b0a02()
667 auto split_dim_size = c->Value(c->Dim(input, split_dim)); in __anon7c94107b0b02() local
674 if (data[i] == -1 && c->ValueKnown(split_dim_size)) { in __anon7c94107b0b02()
675 size = split_dim_size - total_size; in __anon7c94107b0b02()
681 if (c->ValueKnown(split_dim_size)) { in __anon7c94107b0b02()
682 if (has_neg_one ? total_size > split_dim_size in __anon7c94107b0b02()
683 : total_size != split_dim_size) { in __anon7c94107b0b02()
685 "can't split axis of size ", split_dim_size, in __anon7c94107b0b02()
|