Searched refs:copy_size (Results 1 – 6 of 6) sorted by relevance
| /tflite-micro-3.4.0-2.7.6/tensorflow/lite/kernels/internal/reference/ |
| D | concatenation.h | 63 const int copy_size = input_shapes[i]->Dims(axis) * base_inner_size; in Concatenation() local
64 const Scalar* input_ptr = input_data[i] + k * copy_size; in Concatenation()
65 memcpy(output_ptr, input_ptr, copy_size * sizeof(Scalar)); in Concatenation()
66 output_ptr += copy_size; in Concatenation()
115 const int copy_size = input_shapes[i]->Dims(axis) * base_inner_size; in ConcatenationWithScaling() local
116 const uint8_t* input_ptr = input_data[i] + k * copy_size; in ConcatenationWithScaling()
119 memcpy(output_ptr, input_ptr, copy_size); in ConcatenationWithScaling()
123 for (int j = 0; j < copy_size; ++j) { in ConcatenationWithScaling()
131 output_ptr += copy_size; in ConcatenationWithScaling()
|
| /tflite-micro-3.4.0-2.7.6/tensorflow/lite/micro/kernels/ |
| D | pack.cc | 47 int copy_size = 1; in PackImpl() local
49 copy_size *= output_dims->data[i]; in PackImpl()
55 TFLITE_DCHECK_EQ(input_size, copy_size * outer_size); in PackImpl()
63 const T* input_ptr = input_data + copy_size * k; in PackImpl()
64 int loc = k * values_count * copy_size + i * copy_size; in PackImpl()
66 for (int j = 0; j < copy_size; ++j) output_ptr[j] = input_ptr[j]; in PackImpl()
|
| D | unpack.cc | 50 int copy_size = 1; in UnpackImpl() local
52 copy_size *= input_dims->data[i]; in UnpackImpl()
58 TFLITE_DCHECK_EQ(output_size, copy_size * outer_size); in UnpackImpl()
66 T* output_ptr = output_data + copy_size * k; in UnpackImpl()
67 int loc = k * output_count * copy_size + i * copy_size; in UnpackImpl()
69 for (int j = 0; j < copy_size; ++j) output_ptr[j] = input_ptr[j]; in UnpackImpl()
|
| D | split.cc | 60 const int copy_size = output_dims->data[axis] * base_inner_size; in SplitImpl() local
61 T* output_ptr = output_data + k * copy_size; in SplitImpl()
62 for (int j = 0; j < copy_size; ++j) output_ptr[j] = input_ptr[j]; in SplitImpl()
63 input_ptr += copy_size; in SplitImpl()
|
| D | split_v.cc | 63 const int copy_size = in SplitImpl() local
65 T* output_ptr = output_data + k * copy_size; in SplitImpl()
66 for (int j = 0; j < copy_size; ++j) output_ptr[j] = input_ptr[j]; in SplitImpl()
67 input_ptr += copy_size; in SplitImpl()
|
| /tflite-micro-3.4.0-2.7.6/tensorflow/lite/micro/kernels/ceva/ |
| D | ceva_tflm_lib.h | 321 int outer_size, int copy_size,
323 void CEVA_TFLM_PackUnpackImplLoopInitSizes(int* const copy_size,
328 int outer_size, int copy_size,
331 int outer_size, int copy_size,
334 int outer_size, int copy_size,
516 int output_count, int copy_size);
519 int output_count, int copy_size);
|
