Home
last modified time | relevance | path

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

123

/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/testing/
Dmlir_convert.py32 output_tensors, # pylint: disable=unused-argument argument
134 output_tensors, argument
/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.py177 def record_operation(op_type, output_tensors, input_tensors, backward_function, argument
185 def record_operation_backprop_only(op_type, output_tensors, input_tensors, argument
193 def record_operation_forwardprop_only(op_type, output_tensors, input_tensors, argument
/external/federated-compute/fcp/aggregation/protocol/
Dconfiguration.proto58 repeated tensorflow.TensorSpecProto output_tensors = 5; field
/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.h85 std::vector<int> output_tensors; member
/external/tensorflow/tensorflow/distribute/experimental/rpc/proto/
Dtf_rpc_service.proto14 repeated TensorProto output_tensors = 1; field
/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()
/external/federated-compute/fcp/client/engine/
Dcommon.h54 std::vector<tensorflow::Tensor> output_tensors; member
Dtflite_wrapper.h39 std::vector<tensorflow::Tensor> output_tensors; member
Dtflite_wrapper.cc189 OutputTensors output_tensors; in ConstructOutputs() local
/external/tensorflow/tensorflow/lite/delegates/hexagon/
Dhexagon_delegate_kernel.cc125 std::vector<hexagon_nn_tensordef> output_tensors; in Eval() local
245 const TfLiteIntArray* output_tensors) { in BuildGraph()
/external/federated-compute/fcp/aggregation/core/
Done_dim_grouping_aggregator.h74 OutputTensorList output_tensors = std::move(*other_ptr).TakeOutputs(); in MergeWith() local
Dagg_vector_aggregator.h57 OutputTensorList output_tensors = std::move(*other_ptr).TakeOutputs(); in MergeWith() local
/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/tflite-support/tensorflow_lite_support/cc/task/text/nlclassifier/
Dnl_classifier.cc270 const std::vector<const TfLiteTensor*>& output_tensors, in Postprocess()
361 std::vector<const TfLiteTensor*> output_tensors = GetOutputTensors(); in Initialize() local
/external/armnn/python/pyarmnn/examples/image_classification/
Donnx_mobilenetv2.py78 output_tensors = ann.make_output_tensors([output_binding_info]) variable
/external/federated-compute/fcp/client/
Dfl_runner.proto108 map<string, tensorflow.TensorProto> output_tensors = 3; field
/external/tensorflow/tensorflow/lite/python/
Dlite.py1047 def _optimize_tf_model(self, graph_def, input_tensors, output_tensors, argument
1103 def convert(self, graph_def, input_tensors, output_tensors): argument
1994 def _optimize_tf_model(self, graph_def, input_tensors, output_tensors, argument
2380 output_tensors, argument
2561 output_tensors, argument
2592 def from_session(cls, sess, input_tensors, output_tensors): argument
2819 def from_session(cls, sess, input_tensors, output_tensors): argument
Dutil.py262 def _convert_op_hints_if_present(sess, graph_def, output_tensors, argument
273 def freeze_graph(sess, input_tensors, output_tensors): argument
Dconvert.py720 def convert_graphdef(input_data, input_tensors, output_tensors, **kwargs): argument
844 def toco_convert(input_data, input_tensors, output_tensors, *args, **kwargs): argument
/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/engine/
Dnode.py234 def output_tensors(self): member in Node
/external/tensorflow/tensorflow/c/
Dc_api_function.cc90 std::vector<OutputTensor>* output_tensors) in ProcessOutputs()
169 std::vector<tensorflow::OutputTensor> output_tensors; in TF_GraphToFunctionWithControlOutputs() local

123