Searched refs:output_tensor_name (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | eval_const_tensor.cc | 265 const string& output_tensor_name = in ExtractConstantSubgraph() local
276 const_inputs->emplace_back(output_tensor_name, tensor_inferred); in ExtractConstantSubgraph()
277 const_inputs_added.insert(output_tensor_name); in ExtractConstantSubgraph()
284 auto it = cached_values->find(output_tensor_name); in ExtractConstantSubgraph()
286 const_inputs_added.count(output_tensor_name) == 0) { in ExtractConstantSubgraph()
287 const_inputs->emplace_back(output_tensor_name, it->second); in ExtractConstantSubgraph()
288 const_inputs_added.insert(output_tensor_name); in ExtractConstantSubgraph()
343 const string output_tensor_name = in EvaluateConstantTensor() local
357 const_inputs, {output_tensor_name}, &outputs); in EvaluateConstantTensor()
373 (*cached_values)[output_tensor_name] = outputs[0]; in EvaluateConstantTensor()
|
| /external/tensorflow/tensorflow/contrib/predictor/ |
| D | saved_model_predictor_test.py | 70 output_tensor_name = predictor.fetch_tensors['outputs'].name
72 output_tensor_name,
113 output_tensor_name = predictor.fetch_tensors[key].name
115 output_tensor_name,
140 output_tensor_name = predictor.fetch_tensors[key].name
142 output_tensor_name,
|
| D | contrib_estimator_predictor_test.py | 57 output_tensor_name = predictor.fetch_tensors[key].name
59 output_tensor_name,
|
| D | core_estimator_predictor_test.py | 68 output_tensor_name = predictor.fetch_tensors[output_key].name
70 output_tensor_name,
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | identify_prelu.cc | 107 const auto output_tensor_name = add_op->outputs[0]; in Run() local
121 prelu_op->outputs = {output_tensor_name}; in Run()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | function_utils.cc | 90 StringPiece output_tensor_name, in AddFunctionOutputWithUniqueName() argument
102 (*function->mutable_ret())[name] = string(output_tensor_name); in AddFunctionOutputWithUniqueName()
|
| D | function_utils.h | 63 StringPiece output_tensor_name,
|
| /external/tensorflow/tensorflow/cc/tools/ |
| D | freeze_saved_model.cc | 96 for (const string& output_tensor_name : outputs) { in GetReachableNodesAndVariables() local
97 nodes_to_visit.push(GetNodeNameFromTensorName(output_tensor_name)); in GetReachableNodesAndVariables()
|
| /external/tensorflow/tensorflow/cc/saved_model/ |
| D | loader.cc | 127 for (const string& output_tensor_name : output_tensor_names) { in RunOnce() local
128 callable_options.add_fetch(output_tensor_name); in RunOnce()
|
| /external/tensorflow/tensorflow/contrib/session_bundle/ |
| D | bundle_shim_test.cc | 48 const string& output_tensor_name) { in ValidateHalfPlusTwo() argument
58 {{input_tensor_name, input}}, {output_tensor_name}, {}, &outputs)); in ValidateHalfPlusTwo()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | while_v2.py | 885 output_tensor_name = fdef.ret[output_arg_name]
887 output_tensor_name,
|