Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/tpu/kernels/
Dsharding_util_ops_test.cc54 std::vector<Tensor>* output_tensors) { in RunGraph()
87 std::vector<Tensor> output_tensors; in TEST() local
126 std::vector<Tensor> output_tensors; in TEST() local
223 std::vector<Tensor> output_tensors; in TEST_P() local
241 std::vector<Tensor> output_tensors; in TEST_P() local
259 std::vector<Tensor> output_tensors; in TEST_P() local
277 std::vector<Tensor> output_tensors; in TEST_P() local
294 std::vector<Tensor> output_tensors; in TEST_P() local
312 std::vector<Tensor> output_tensors; in TEST_P() local
329 std::vector<Tensor> output_tensors; in TEST_P() local
[all …]
/external/tensorflow/tensorflow/lite/experimental/mlir/testing/
Dmlir_convert.py30 def mlir_convert(options, graph_def, input_tensors, output_tensors, **kwargs): argument
127 output_tensors, argument
/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/delegates/gpu/cl/testing/
Dinternal_api_samples.cc206 std::vector<std::vector<float>> output_tensors(out_refs.size()); in RunModelSampleWithInternalAPI() local
331 std::vector<std::vector<float>> output_tensors(out_refs.size()); in RunModelSampleWithInternalAPISerializedKernels() local
412 std::vector<std::vector<float>> output_tensors(out_refs.size()); in RunModelSampleWithInternalAPISerialized() local
/external/tensorflow/tensorflow/lite/
Dgraph_info.h83 std::vector<int> output_tensors; member
/external/tensorflow/tensorflow/lite/testing/
Dtoco_convert.py76 def toco_convert(options, graph_def, input_tensors, output_tensors, **kwargs): argument
/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()
/external/tensorflow/tensorflow/lite/python/
Dconvert.py497 output_tensors, argument
774 def toco_convert_impl(input_data, input_tensors, output_tensors, argument
839 def toco_convert(input_data, input_tensors, output_tensors, *args, **kwargs): argument
Dlite.py863 def _optimize_tf_model(self, graph_def, input_tensors, output_tensors, argument
889 def convert(self, graph_def, input_tensors, output_tensors): argument
1643 def _optimize_tf_model(self, graph_def, input_tensors, output_tensors, argument
2029 output_tensors, argument
2208 output_tensors, argument
2239 def from_session(cls, sess, input_tensors, output_tensors): argument
2453 def from_session(cls, sess, input_tensors, output_tensors): argument
Dutil.py254 def _convert_op_hints_if_present(sess, graph_def, output_tensors, argument
265 def freeze_graph(sess, input_tensors, output_tensors): argument
/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/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/core/grappler/utils/
Dgrappler_test.cc122 std::vector<Tensor> output_tensors; in EvaluateNodes() local
139 std::vector<Tensor> output_tensors; in EvaluateFetchNodes() local
/external/armnn/python/pyarmnn/examples/image_classification/
Donnx_mobilenetv2.py78 output_tensors = ann.make_output_tensors([output_binding_info]) variable
/external/ComputeLibrary/tests/validation/reference/
DUnstack.cpp87 …nsor<T>> unstack(const SimpleTensor<T> &input_tensor, std::vector<SimpleTensor<T>> &output_tensors in unstack()
/external/tensorflow/tensorflow/python/keras/feature_column/
Dbase_feature_layer.py117 def _verify_and_concat_tensors(self, output_tensors): argument
/external/tensorflow/tensorflow/python/keras/engine/
Dnode.py234 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/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/core/kernels/data/experimental/
Dio_ops.cc183 std::vector<Tensor> output_tensors; in GetShardIndex() local
350 std::vector<Tensor> output_tensors; in GetNextInternal() local
455 std::vector<Tensor> output_tensors; in GetShardIndex() local
/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.cc264 std::vector<Tensor> output_tensors; in VerifyFusedBatchNorm() local
401 std::vector<Tensor> output_tensors; in VerifyFusedBatchNormGradWithConv2D() local
/external/tensorflow/tensorflow/lite/delegates/
Dserialization.cc274 auto* output_tensors = delegate_params->output_tensors; in GetEntryImpl() local

12