Searched refs:VectorOfTensors (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | portable_tensor.h | 33 class VectorOfTensors {
36 VectorOfTensors(const TfLiteContext& context, in VectorOfTensors() function
77 class VectorOfQuantizedTensors : public VectorOfTensors<uint8_t> {
82 : VectorOfTensors<uint8_t>(context, tensor_list) { in VectorOfQuantizedTensors()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | concatenation.cc | 91 VectorOfTensors<int8_t> all_inputs(*context, *node->inputs); in Prepare()
128 VectorOfTensors<scalar> all_inputs(*context, *node->inputs); \ in Eval()
|
| D | add_n.cc | 62 VectorOfTensors<T> all_inputs(*context, *node->inputs); in EvalAddN()
|
| D | split.cc | 131 VectorOfTensors<scalar> all_outputs(*context, *node->outputs); \ in Eval()
|
| D | unpack.cc | 93 VectorOfTensors<T> all_outputs(*context, *node->outputs); in UnpackImpl()
|
| D | pack.cc | 100 VectorOfTensors<T> all_inputs(*context, *node->inputs); in PackImpl()
|
| D | split_v.cc | 175 VectorOfTensors<scalar> all_outputs(*context, *node->outputs); \ in Eval()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | add_n.cc | 62 VectorOfTensors<T> all_inputs(*context, *node->inputs); in EvalAddN()
|