Searched defs:split_size (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | split_op.cc | 240 auto make_sizes = [&](Eigen::DenseIndex split_size) { in Compute()
243 auto reshape_result = [&](Tensor* result, Eigen::DenseIndex split_size) { in Compute()
253 auto make_sizes = [&](Eigen::DenseIndex split_size) { in Compute()
257 auto reshape_result = [&](Tensor* result, Eigen::DenseIndex split_size) { in Compute()
|
| D | split_v_op.cc | 302 auto make_sizes = [&](Eigen::DenseIndex split_size) { in Compute()
305 auto reshape_result = [&](Tensor* result, Tlen split_size) { in Compute()
315 auto make_sizes = [&](Eigen::DenseIndex split_size) { in Compute()
319 auto reshape_result = [&](Tensor* result, Tlen split_size) { in Compute()
|
| D | concat_lib_gpu_impl.cu.cc | 38 GpuDeviceArrayStruct<const T*> input_ptr_data, int split_size, in concat_fixed_kernel()
141 bool fixed_size, int split_size, in ConcatGPUImpl()
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | sparse_tensor.h | 329 static inline int GetSliceIndex(const int dim, const int split_size, in GetSliceIndex()
343 static inline int GetDimensionInSlice(const int dim, const int split_size, in GetDimensionInSlice()
357 static inline int GetSliceShape(const int slice_index, const int split_size, in GetSliceShape()
592 const int split_size = split_dim_size / num_split; in Split() local
|
| /external/libaom/libaom/av1/decoder/ |
| D | decodeframe.c | 1244 int split_size, int txs, int blk_row, in set_inter_tx_size()
|