Searched defs:split_size (Results 1 – 15 of 15) sorted by relevance
| /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 | 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()
|
| D | ragged_tensor_from_variant_op.cc | 117 int split_size = 1; in NestedStackRaggedTensors() local
|
| D | ragged_tensor_to_variant_op.cc | 84 int split_size = -1; in UnbatchRaggedZerothDim() 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 split_size = in ExpandInstruction() local
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | split_v.cc | 40 int64_t split_size = 0; in SplitImpl() local
|
| D | split.cc | 42 int64_t split_size = output_dims->data[axis] * output_count; in SplitImpl() local
|
| D | split_v_test.cc | 54 constexpr int split_size = 1; in TestSplitVFloat() local
|
| /external/libgav1/libgav1/src/tile/ |
| D | tile.cc | 2274 const BlockSize split_size = kSubSize[kPartitionSplit][block_size]; in ProcessPartition() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 842 int64_t split_size = 0; in Split() local
|
| /external/libaom/libaom/av1/decoder/ |
| D | decodeframe.c | 1022 int min_txs, int split_size, int txs, in set_inter_tx_size()
|