| /external/ComputeLibrary/src/graph/nodes/ |
| D | SplitLayerNode.cpp | 60 int split_size = input_descriptor.shape[tmp_axis] / num_splits; in compute_output_descriptor() local
63 output_descriptor.shape.set(tmp_axis, split_size); in compute_output_descriptor()
64 coords.set(tmp_axis, idx * split_size); in compute_output_descriptor()
68 int split_size = _size_splits[idx]; in compute_output_descriptor() local
69 if(split_size == -1) in compute_output_descriptor()
71 split_size = input_descriptor.shape[tmp_axis]; in compute_output_descriptor()
73 split_size -= _size_splits[i]; in compute_output_descriptor()
75 output_descriptor.shape.set(tmp_axis, split_size); in compute_output_descriptor()
119 …int split_size = (_size_splits.empty()) ? (input_descriptor.shape[tmp_axis] / _num_splits) : _size… in configure_output() local
120 if(split_size == -1) in configure_output()
[all …]
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | sparse_tensor.h | 221 static inline int GetSliceIndex(const int dim, const int split_size, in GetSliceIndex() argument
223 DCHECK_GT(split_size, 0); in GetSliceIndex()
225 if (residual == 0) return dim / split_size; in GetSliceIndex()
226 const int offset = residual * (split_size + 1); in GetSliceIndex()
228 return dim / (split_size + 1); in GetSliceIndex()
230 return residual + ((dim - offset) / split_size); in GetSliceIndex()
235 static inline int GetDimensionInSlice(const int dim, const int split_size, in GetDimensionInSlice() argument
237 DCHECK_GT(split_size, 0); in GetDimensionInSlice()
239 if (residual == 0) return dim % split_size; in GetDimensionInSlice()
240 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 config.thread_per_block, 0, gpu_device.stream(), input_ptrs, split_size, in ConcatGPUImpl()
195 int split_size, typename TTypes<T, 2>::Matrix* output);
202 int split_size, typename TTypes<T, 2>::Matrix* output);
|
| D | split_op.cc | 237 auto make_sizes = [&](Eigen::DenseIndex split_size) { in Compute() argument
238 return Eigen::DSizes<Eigen::DenseIndex, 2>{split_size, suffix_dim_size}; in Compute()
240 auto reshape_result = [&](Tensor* result, Eigen::DenseIndex split_size) { in Compute() argument
241 return result->shaped<T, 2>({split_size, suffix_dim_size}); in Compute()
250 auto make_sizes = [&](Eigen::DenseIndex split_size) { in Compute() argument
251 return Eigen::DSizes<Eigen::DenseIndex, 3>{prefix_dim_size, split_size, in Compute()
254 auto reshape_result = [&](Tensor* result, Eigen::DenseIndex split_size) { in Compute() argument
256 {prefix_dim_size, split_size, suffix_dim_size}); in Compute()
|
| D | split_v_op.cc | 197 for (const Tlen split_size : split_sizes) { in SplitHasAlignedOutputsInFirstDimension() local
198 if (!IsDim0SliceAligned<T>(input_shape, start, start + split_size)) { in SplitHasAlignedOutputsInFirstDimension()
201 start += split_size; in SplitHasAlignedOutputsInFirstDimension()
391 auto make_sizes = [&](Eigen::DenseIndex split_size) { in Compute() argument
392 return Eigen::DSizes<Eigen::DenseIndex, 2>{split_size, suffix_dim_size}; in Compute()
394 auto reshape_result = [&](Tensor* result, Tlen split_size) { in Compute() argument
395 return result->shaped<T, 2>({split_size, suffix_dim_size}); in Compute()
404 auto make_sizes = [&](Eigen::DenseIndex split_size) { in Compute() argument
405 return Eigen::DSizes<Eigen::DenseIndex, 3>{prefix_dim_size, split_size, in Compute()
408 auto reshape_result = [&](Tensor* result, Tlen split_size) { in Compute() argument
[all …]
|
| D | ragged_tensor_to_variant_op.cc | 84 int split_size = -1; in UnbatchRaggedZerothDim() local
87 split_size = in UnbatchRaggedZerothDim()
92 split_size = ragged_component_splits_vec[j - 1](last_index) + 1; in UnbatchRaggedZerothDim()
95 Tensor(DataTypeToEnum<SPLIT_TYPE>::value, TensorShape({split_size}))); in UnbatchRaggedZerothDim()
100 for (int k = 1; k < split_size; k++, index[j]++) { in UnbatchRaggedZerothDim()
|
| 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);
|
| D | ragged_tensor_from_variant_op.cc | 117 int split_size = 1; in NestedStackRaggedTensors() local
120 split_size += ragged_components[j].splits(i).NumElements() - 1; in NestedStackRaggedTensors()
124 Tensor(DataTypeToEnum<SPLIT_TYPE>::value, TensorShape({split_size}))); in NestedStackRaggedTensors()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | all_to_all_decomposer.cc | 63 int64_t split_size = in ExpandInstruction() local
75 new_all_to_all_shape.add_dimensions(split_size); in ExpandInstruction()
99 slice_limits[split_dim] = split_size; in ExpandInstruction()
101 slice_shape.set_dimensions(split_dim, split_size); in ExpandInstruction()
111 slice_limits[split_dim] += split_size; in ExpandInstruction()
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/kernel/sentencepiece/ |
| D | sentencepiece_detokenizer_tflite.cc | 72 const int split_size = input_splits_data[i + 1] - input_splits_data[i]; in Eval() local
75 input_encoded_data + input_offset + split_size, in Eval()
81 input_offset += split_size; in Eval()
|
| D | sentencepiece_detokenizer_op.cc | 64 const int split_size = input_splits_flat(i + 1) - input_splits_flat(i); in Compute() local
66 codes_for_split.reserve(split_size); in Compute()
67 for (int j = 0; j < split_size; ++j) { in Compute()
|
| /external/tensorflow/tensorflow/security/advisory/ |
| D | tfsa-2021-046.md | 28 int slice_index = GetSliceIndex(dim, split_size, residual);
|
| /external/libgav1/libgav1/src/tile/ |
| D | tile.cc | 2320 const BlockSize split_size = kSubSize[kPartitionSplit][block_size]; in ProcessPartition() local
2357 if (!ProcessBlock(row4x4, column4x4, split_size, scratch_buffer, in ProcessPartition()
2359 !ProcessBlock(row4x4, column4x4 + half_block4x4, split_size, in ProcessPartition()
2369 !ProcessBlock(row4x4 + half_block4x4, column4x4, split_size, in ProcessPartition()
2372 split_size, scratch_buffer, residual)) { in ProcessPartition()
2377 if (!ProcessBlock(row4x4, column4x4, split_size, scratch_buffer, in ProcessPartition()
2379 !ProcessBlock(row4x4 + half_block4x4, column4x4, split_size, in ProcessPartition()
2389 !ProcessBlock(row4x4, column4x4 + half_block4x4, split_size, in ProcessPartition()
2392 split_size, scratch_buffer, residual)) { in ProcessPartition()
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_array_ops.py | 198 split_size = math_ops.cumprod(mask_shape) + 1
201 masked_splits = math_ops.range(split_size[dim]) * elt_size
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_ops.py | 766 split_size = total_grad_size // num_splits
767 split_size_last = total_grad_size - split_size * (num_splits - 1)
768 split_sizes = [split_size] * (num_splits - 1) + [split_size_last]
|
| D | tpu_strategy.py | 534 split_size = partition_dimensions[dim_index]
535 if dim_size % split_size != 0:
539 dim_index, dim_size, split_size))
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_functional_ops.cc | 1767 int split_size = 0; in ReplaceAndPartitionXLAShardingVariable() local
1779 split_size = xla_sharding.tile_assignment_dimensions(dim); in ReplaceAndPartitionXLAShardingVariable()
1808 if (dim_size % split_size != 0) { in ReplaceAndPartitionXLAShardingVariable()
1811 split_size); in ReplaceAndPartitionXLAShardingVariable()
1813 proto.mutable_dim(split_dim)->set_size(dim_size / split_size); in ReplaceAndPartitionXLAShardingVariable()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 844 int64_t split_size = 0; in Split() local
852 split_size += output_shapes[i]->Dims(axis); in Split()
854 TFLITE_DCHECK_EQ(split_size, input_shape.Dims(axis)); in Split()
|
| /external/libaom/libaom/av1/decoder/ |
| D | decodeframe.c | 1012 int min_txs, int split_size, int txs, in set_inter_tx_size() argument
1014 for (int idy = 0; idy < tx_size_high_unit[split_size]; in set_inter_tx_size()
1016 for (int idx = 0; idx < tx_size_wide_unit[split_size]; in set_inter_tx_size()
|