Home
last modified time | relevance | path

Searched defs:output_tensors (Results 1 – 25 of 44) sorted by relevance

12

/external/tensorflow/tensorflow/lite/micro/kernels/
Dsplit_v_test.cc37 const OutputTensors<N>& output_tensors) { in TestSplitVFloat()
126 tflite::testing::OutputTensors<3> output_tensors; in TF_LITE_MICRO_TEST() local
162 tflite::testing::OutputTensors<2> output_tensors; in TF_LITE_MICRO_TEST() local
196 tflite::testing::OutputTensors<2> output_tensors; in TF_LITE_MICRO_TEST() local
230 tflite::testing::OutputTensors<2> output_tensors; in TF_LITE_MICRO_TEST() local
263 tflite::testing::OutputTensors<2> output_tensors; in TF_LITE_MICRO_TEST() local
297 tflite::testing::OutputTensors<2> output_tensors; in TF_LITE_MICRO_TEST() local
357 tflite::testing::OutputTensors<8> output_tensors; in TF_LITE_MICRO_TEST() local
434 tflite::testing::OutputTensors<8> output_tensors; in TF_LITE_MICRO_TEST() local
/external/tensorflow/tensorflow/python/eager/
Dtape.py181 def record_operation(op_type, output_tensors, input_tensors, backward_function, argument
189 def record_operation_backprop_only(op_type, output_tensors, input_tensors, argument
197 def record_operation_forwardprop_only(op_type, output_tensors, input_tensors, argument
/external/tensorflow/tensorflow/lite/
Dgraph_info.h83 std::vector<int> output_tensors; member
/external/tensorflow/tensorflow/lite/testing/
Dtoco_convert.py78 def toco_convert(options, graph_def, input_tensors, output_tensors, **kwargs): argument
/external/tensorflow/tensorflow/lite/python/
Dconvert.py364 output_tensors, argument
599 def toco_convert_impl(input_data, input_tensors, output_tensors, argument
662 def toco_convert(input_data, input_tensors, output_tensors, *args, **kwargs): argument
Dutil.py292 def _convert_op_hints_if_present(sess, graph_def, output_tensors, argument
303 def freeze_graph(sess, input_tensors, output_tensors): argument
Dlite.py607 def convert(self, graph_def, input_tensors, output_tensors): argument
1631 output_tensors, argument
1791 output_tensors, argument
1822 def from_session(cls, sess, input_tensors, output_tensors): argument
2036 def from_session(cls, sess, input_tensors, output_tensors): argument
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/testing/
Dinternal_api_samples.cc206 std::vector<std::vector<float>> output_tensors(out_refs.size()); in RunModelSampleWithInternalAPI() local
330 std::vector<std::vector<float>> output_tensors(out_refs.size()); in RunModelSampleWithInternalAPISerializedKernels() local
406 std::vector<std::vector<float>> output_tensors(out_refs.size()); in RunModelSampleWithInternalAPISerialized() local
/external/tflite-support/tensorflow_lite_support/cc/task/vision/
Dobject_detector.cc163 const std::vector<const TfLiteTensor*>& output_tensors) { in SanityCheckOutputTensors()
447 const std::vector<const TfLiteTensor*>& output_tensors, in Postprocess()
Dimage_segmenter.cc315 const std::vector<const TfLiteTensor*>& output_tensors, in Postprocess()
Dimage_classifier.cc404 const std::vector<const TfLiteTensor*>& output_tensors, in Postprocess()
/external/tflite-support/tensorflow_lite_support/cc/task/text/nlclassifier/
Dnl_classifier.cc244 const std::vector<const TfLiteTensor*>& output_tensors, in Postprocess()
335 std::vector<const TfLiteTensor*> output_tensors = GetOutputTensors(); in Initialize() local
Dbert_nl_classifier.cc119 const std::vector<const TfLiteTensor*>& output_tensors, in Postprocess()
/external/tensorflow/tensorflow/lite/delegates/hexagon/
Dhexagon_delegate_kernel.cc123 std::vector<hexagon_nn_tensordef> output_tensors; in Eval() local
241 const TfLiteIntArray* output_tensors) { in BuildGraph()
/external/tensorflow/tensorflow/core/grappler/utils/
Dgrappler_test.cc122 std::vector<Tensor> output_tensors; in EvaluateNodes() local
139 std::vector<Tensor> output_tensors; in EvaluateFetchNodes() local
/external/tensorflow/tensorflow/python/keras/feature_column/
Dbase_feature_layer.py115 def _verify_and_concat_tensors(self, output_tensors): argument
/external/tensorflow/tensorflow/python/keras/engine/
Dnode.py237 def output_tensors(self): member in Node
/external/tensorflow/tensorflow/c/
Dc_api_function.cc89 std::vector<OutputTensor>* output_tensors) in ProcessOutputs()
168 std::vector<tensorflow::OutputTensor> output_tensors; in TF_GraphToFunctionWithControlOutputs() local
/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_utils_test.cc234 std::vector<tensorflow::Tensor> output_tensors; in TEST() local
254 std::vector<tensorflow::Tensor> output_tensors; in TEST() local
280 std::vector<tensorflow::Tensor> output_tensors; in TEST() local
Dremote_fused_graph_execute_op_test.cc291 std::vector<Tensor> output_tensors; in TEST() local
/external/tensorflow/tensorflow/c/eager/
Dtape.h407 const string& op_type, const std::vector<TapeTensor>& output_tensors, in RecordOperation()
925 const string& op_type, const std::vector<TapeTensor>& output_tensors, in ForwardpropFromTape()
1024 const std::vector<TapeTensor>& output_tensors, in Accumulate()
/external/tensorflow/tensorflow/cc/framework/
Dwhile_gradients.cc36 const std::vector<OutputTensor>& output_tensors) { in ToOutputVector()
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_fused_batch_norm_op_test.cc263 std::vector<Tensor> output_tensors; in VerifyFusedBatchNorm() local
398 std::vector<Tensor> output_tensors; in VerifyFusedBatchNormGradWithConv2D() local
/external/tensorflow/tensorflow/tools/benchmark/
Dbenchmark_model.cc122 std::vector<tensorflow::Tensor> output_tensors; in GetOutputShapes() local
291 std::vector<tensorflow::Tensor> output_tensors; in RunBenchmark() local
/external/tflite-support/tensorflow_lite_support/cc/task/text/qa/
Dbert_question_answerer.cc287 const std::vector<const TfLiteTensor*>& output_tensors, in Postprocess()

12