Searched refs:SequentialTensorWriter (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/ |
| H A D | tensor.h | 26 class SequentialTensorWriter<string> {
28 SequentialTensorWriter(const TfLiteTensor* input, TfLiteTensor* output) in SequentialTensorWriter() function
30 ~SequentialTensorWriter() { buffer_.WriteToTensor(output_, nullptr); } in ~SequentialTensorWriter()
|
| H A D | portable_tensor.h | 100 class SequentialTensorWriter {
102 SequentialTensorWriter(const TfLiteTensor* input, TfLiteTensor* output) { in SequentialTensorWriter() function
106 SequentialTensorWriter(const T* input_data, T* output_data) in SequentialTensorWriter() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | slice.h | 29 SequentialTensorWriter<T>* writer) { in Slice()
65 SequentialTensorWriter<T> writer(input_data, output_data); in Slice()
73 SequentialTensorWriter<T> writer(input, output); in Slice()
|
| H A D | strided_slice.h | 33 SequentialTensorWriter<T>* writer) { in StridedSlice()
102 SequentialTensorWriter<T> writer(input_data, output_data); in StridedSlice()
113 SequentialTensorWriter<T> writer(input, output); in StridedSlice()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | roll.cc | 65 SequentialTensorWriter<T> writer(output, cache); in Pool()
69 SequentialTensorWriter<T> writer(source, output); in Pool()
86 SequentialTensorWriter<T> writer(input, output); in Pool()
|
| H A D | squeeze.cc | 86 SequentialTensorWriter<string> writer(op_context.input, op_context.output); in Eval()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| H A D | optimized_ops.h | 4627 SequentialTensorWriter<T>* writer) { in Slice()
4664 SequentialTensorWriter<T> writer(input_data, output_data); in Slice()
4672 SequentialTensorWriter<T> writer(input, output); in Slice()
4682 SequentialTensorWriter<T>* writer) { in StridedSlice()
4762 SequentialTensorWriter<T> writer(input_data, output_data); in StridedSlice()
4773 SequentialTensorWriter<T> writer(input, output); in StridedSlice()
|