Searched refs:SequentialTensorWriter (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| 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()
|
| 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
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| 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()
|
| D | reference_ops.h | 1373 SequentialTensorWriter<T>* writer) { in Slice()
1409 SequentialTensorWriter<T> writer(input_data, output_data); in Slice()
1417 SequentialTensorWriter<T> writer(input, output); in Slice()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | squeeze.cc | 86 SequentialTensorWriter<string> writer(op_context.input, op_context.output); in Eval()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 5389 SequentialTensorWriter<T>* writer) { in Slice()
5426 SequentialTensorWriter<T> writer(input_data, output_data); in Slice()
5434 SequentialTensorWriter<T> writer(input, output); in Slice()
5444 SequentialTensorWriter<T>* writer) { in StridedSlice()
5524 SequentialTensorWriter<T> writer(input_data, output_data); in StridedSlice()
5535 SequentialTensorWriter<T> writer(input, output); in StridedSlice()
|