Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/learn/python/learn/utils/
Dsaved_model_export_utils.py80 def build_standardized_signature_def(input_tensors, output_tensors, argument
129 def _get_classification_scores(output_tensors): argument
136 def _get_classification_classes(output_tensors): argument
144 def _is_classification_problem(problem_type, input_tensors, output_tensors): argument
154 def _is_regression_problem(problem_type, input_tensors, output_tensors): argument
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Dmodel_fn.py216 def _scores(output_tensors): argument
222 def _classes(output_tensors): # pylint: disable=missing-docstring argument
/external/tensorflow/tensorflow/core/grappler/utils/
Dgrappler_test.cc31 std::vector<Tensor> output_tensors; in EvaluateNodes() local
/external/tensorflow/tensorflow/contrib/lite/
Dgraph_info.h67 std::vector<int> output_tensors; member
/external/tensorflow/tensorflow/python/eager/
Dtape.py79 def record_operation(op_type, output_tensors, input_tensors, backward_function): argument
Dpywrap_tfe_src.cc1065 void TFE_Py_TapeSetRecordOperation(PyObject* op_type, PyObject* output_tensors,
/external/tensorflow/tensorflow/contrib/lite/python/
Dlite.py143 output_tensors, argument
/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
Dremote_fused_graph_execute_utils.cc203 std::vector<tensorflow::Tensor>* output_tensors) { in DryRunInference()
271 std::vector<Tensor> output_tensors; in DryRunInferenceForAllNode() local
337 const std::vector<tensorflow::Tensor>& output_tensors, in ConvertToTensorShapeMap()
/external/tensorflow/tensorflow/cc/framework/
Dwhile_gradients.cc36 const std::vector<OutputTensor>& output_tensors) { in ToOutputVector()
/external/tensorflow/tensorflow/c/
Dc_api_function.cc426 std::vector<OutputTensor>* output_tensors) in ProcessOutputs()
505 std::vector<tensorflow::OutputTensor> output_tensors; in TF_GraphToFunction() local
/external/tensorflow/tensorflow/tools/benchmark/
Dbenchmark_model.cc140 std::vector<tensorflow::Tensor> output_tensors; in GetOutputShapes() local
256 std::vector<tensorflow::Tensor> output_tensors; in RunBenchmark() local
/external/tensorflow/tensorflow/python/layers/
Dbase.py819 output_tensors, argument
1330 output_tensors, argument
/external/tensorflow/tensorflow/c/eager/
Druntime.cc264 std::vector<Tensor>* output_tensors, in Run()
Dtape.h189 const string& op_type, gtl::ArraySlice<TapeTensor> output_tensors, in RecordOperation()
/external/tensorflow/tensorflow/core/util/sparse/
Dsparse_tensor.h559 std::vector<SparseTensor> output_tensors; in Split() local
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dhexagon_graph_execution_test.cc290 std::vector<Tensor> output_tensors; in RunFusedGraph() local
/external/tensorflow/tensorflow/contrib/lite/testing/
Dgenerate_examples.py292 def toco_convert(graph_def_str, input_tensors, output_tensors, argument
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding.cc896 TensorVector output_tensors; in EvaluateOneFoldable() local