| /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_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 | split_v_op.cc | 197 for (const Tlen split_size : split_sizes) { in SplitHasAlignedOutputsInFirstDimension() local
391 auto make_sizes = [&](Eigen::DenseIndex split_size) { in Compute()
394 auto reshape_result = [&](Tensor* result, Tlen split_size) { in Compute()
404 auto make_sizes = [&](Eigen::DenseIndex split_size) { in Compute()
408 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()
|
| D | ragged_tensor_to_variant_op.cc | 84 int split_size = -1; in UnbatchRaggedZerothDim() local
|
| D | ragged_tensor_from_variant_op.cc | 117 int split_size = 1; in NestedStackRaggedTensors() local
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | sparse_tensor.h | 221 static inline int GetSliceIndex(const int dim, const int split_size, in GetSliceIndex()
235 static inline int GetDimensionInSlice(const int dim, const int split_size, in GetDimensionInSlice()
249 static inline int GetSliceShape(const int slice_index, const int split_size, in GetSliceShape()
515 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/libgav1/libgav1/src/tile/ |
| D | tile.cc | 2320 const BlockSize split_size = kSubSize[kPartitionSplit][block_size]; in ProcessPartition() local
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_functional_ops.cc | 1767 int split_size = 0; in ReplaceAndPartitionXLAShardingVariable() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 844 int64_t split_size = 0; in Split() local
|
| /external/libaom/libaom/av1/decoder/ |
| D | decodeframe.c | 1012 int min_txs, int split_size, int txs, in set_inter_tx_size()
|