Home
last modified time | relevance | path

Searched refs:output_tensor (Results 1 – 11 of 11) sorted by relevance

/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/kernels/
A Dconcatenation.cc40 inline int CalculatePositiveAxis(int axis, const TfLiteTensor* output_tensor) { in CalculatePositiveAxis() argument
44 return NumDimensions(output_tensor) + axis; in CalculatePositiveAxis()
142 const TfLiteTensor* output_tensor = GetOutput(context, node, kOutputTensor); in Prepare() local
143 TF_LITE_ENSURE(context, output_tensor != nullptr); in Prepare()
144 TfLiteType output_type = output_tensor->type; in Prepare()
230 const TfLiteTensor* output_tensor = GetOutput(context, node, kOutputTensor); in Eval() local
231 TF_LITE_ENSURE(context, output_tensor != nullptr); in Eval()
232 TfLiteType output_type = output_tensor->type; in Eval()
A Dsplit_v.cc60 TfLiteEvalTensor* output_tensor = in SplitImpl() local
62 T* output_data = tflite::micro::GetTensorData<T>(output_tensor); in SplitImpl()
64 output_tensor->dims->data[axis_value] * base_inner_size; in SplitImpl()
A Dsvdf.h57 TfLiteEvalTensor* output_tensor,
A Dsvdf_common.cc58 TfLiteEvalTensor* output_tensor, in EvalIntegerSvdfReference() argument
190 tflite::micro::GetTensorData<int8_t>(output_tensor)[i] = in EvalIntegerSvdfReference() local
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/experimental/mlir/testing/op_tests/
A Dparse_example.py78 output_tensor = out["x"]
80 output_tensor = out["x"].values
81 return [input_value], [output_tensor]
/AliOS-Things-master/solutions/tflite_micro_speech_demo/micro_speech/train/speech_commands/
A Dfreeze.py176 def save_saved_model(file_name, sess, input_tensor, output_tensor): argument
191 'output': tf.compat.v1.saved_model.utils.build_tensor_info(output_tensor)
225 input_tensor, output_tensor = create_inference_graph(
240 save_saved_model(FLAGS.output_file, sess, input_tensor, output_tensor)
A Dfreeze_test.py115 input_tensor, output_tensor = freeze.create_inference_graph(
129 output_tensor)
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/
A Dmicro_interpreter.h97 TfLiteTensor* output_tensor(size_t index) { return output(index); } in output_tensor() function
100 if (TfLiteTensor* tensor_ptr = output_tensor(tensor_index)) { in typed_output_tensor()
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/kernels/cmsis_nn/
A Dsvdf.cc213 TfLiteEvalTensor* output_tensor, const OpData& data) { in EvalIntegerSVDF() argument
234 output_dims.n = output_tensor->dims->data[0]; in EvalIntegerSVDF()
235 output_dims.h = output_tensor->dims->data[1]; in EvalIntegerSVDF()
267 int8_t* output_data = tflite::micro::GetTensorData<int8_t>(output_tensor); in EvalIntegerSVDF()
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/c/
A Dc_api.h281 const TfLiteTensor* output_tensor, void* output_data,
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/
A Dinterpreter.h384 TfLiteTensor* output_tensor(size_t index) { return tensor(outputs()[index]); } in output_tensor() function
388 const TfLiteTensor* output_tensor(size_t index) const { in output_tensor() function

Completed in 12 milliseconds