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