Home
last modified time | relevance | path

Searched defs:output_tensors (Results 1 – 25 of 50) 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/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/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()
Dimage_segmenter.cc315 const std::vector<const TfLiteTensor*>& output_tensors, in Postprocess()
/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/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
Dbert_nl_classifier.cc129 const std::vector<const TfLiteTensor*>& output_tensors, in Postprocess()
/external/armnn/python/pyarmnn/examples/image_classification/
Donnx_mobilenetv2.py78 output_tensors = ann.make_output_tensors([output_binding_info]) variable
/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
/external/tensorflow/tensorflow/compiler/jit/tests/
Dxla_compilation_cache_test_helper.cc107 std::vector<Tensor> output_tensors; in ExecuteWithBatch() 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.cc185 std::vector<Tensor> output_tensors; in GetShardIndex() local
352 std::vector<Tensor> output_tensors; in GetNextInternal() local
457 std::vector<Tensor> output_tensors; in GetShardIndex() local
/external/ComputeLibrary/src/dynamic_fusion/sketch/gpu/
DGpuKernelComponentGroup.cpp136 std::set<const ITensorInfo *> output_tensors; in finalize() local
/external/tensorflow/tensorflow/cc/framework/
Dwhile_gradients.cc36 const std::vector<OutputTensor>& output_tensors) { in ToOutputVector()

12