| /external/ComputeLibrary/src/graph/nodes/ |
| D | SplitLayerNode.cpp | 60 int split_size = input_descriptor.shape[tmp_axis] / num_splits; in compute_output_descriptor() local
68 int split_size = _size_splits[idx]; in compute_output_descriptor() local
119 …int split_size = (_size_splits.empty()) ? (input_descriptor.shape[tmp_axis] / _num_splits) : _size… in configure_output() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | split_v_op.cc | 142 const Tlen& split_size = (*split_sizes_vec)[i]; in ComputeEasyCases() local
204 for (const Tlen split_size : split_sizes) { in SplitHasAlignedOutputsInFirstDimension() local
337 auto make_sizes = [&](Eigen::DenseIndex split_size) { in Compute()
340 auto reshape_result = [&](Tensor* result, Tlen split_size) { in Compute()
350 auto make_sizes = [&](Eigen::DenseIndex split_size) { in Compute()
354 auto reshape_result = [&](Tensor* result, Tlen split_size) { in Compute()
|
| D | split_op.cc | 237 auto make_sizes = [&](Eigen::DenseIndex split_size) { in Compute()
240 auto reshape_result = [&](Tensor* result, Eigen::DenseIndex split_size) { in Compute()
250 auto make_sizes = [&](Eigen::DenseIndex split_size) { in Compute()
254 auto reshape_result = [&](Tensor* result, Eigen::DenseIndex 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()
|
| D | sparse_split_op_gpu.cu.cc | 49 inline __device__ Index GetSliceIndex(const Index index, const Index split_size, in GetSliceIndex()
62 const Index split_size, in GetDimensionInSlice()
74 inline Index GetSliceShape(const Index slice_index, const Index split_size, in GetSliceShape()
|
| D | ragged_tensor_to_variant_op.cc | 125 SPLIT_TYPE split_size = -1; in UnbatchRaggedZerothDim() local
|
| D | ragged_tensor_from_variant_op.cc | 174 int split_size = 1; in NestedStackRaggedTensors() local
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | sparse_tensor.h | 222 static inline int GetSliceIndex(const int dim, const int split_size, in GetSliceIndex()
236 static inline int GetDimensionInSlice(const int dim, const int split_size, in GetDimensionInSlice()
250 static inline int GetSliceShape(const int slice_index, const int split_size, in GetSliceShape()
516 const int split_size = split_dim_size / num_split; in Split() local
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/kernel/sentencepiece/ |
| D | sentencepiece_detokenizer_op.cc | 64 const int split_size = input_splits_flat(i + 1) - input_splits_flat(i); in Compute() local
|
| D | sentencepiece_detokenizer_tflite.cc | 72 const int split_size = input_splits_data[i + 1] - input_splits_data[i]; in Eval() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | all_to_all_decomposer.cc | 63 int64_t split_size = in ExpandInstruction() local
|
| /external/crosvm/swap/src/ |
| D | page_handler.rs | 436 let split_size = if chunk_size >= hugepage_size { in move_to_staging() localVariable
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 467 int64_t split_size = 0; in Split() local
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_functional_ops.cc | 1822 int split_size = 0; in ReplaceAndPartitionXLAShardingVariable() local
|
| /external/libgav1/src/tile/ |
| D | tile.cc | 2320 const BlockSize split_size = kSubSize[kPartitionSplit][block_size]; in ProcessPartition() local
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate.cc | 1344 int split_size = size_splits[split_index] == -1 in TransformSplitVIntoSupportedOps() local
|
| /external/libaom/av1/decoder/ |
| D | decodeframe.c | 1003 int min_txs, int split_size, int txs, in set_inter_tx_size()
|