Searched refs:cached_values (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/core/common_runtime/ |
D | eval_const_tensor.cc | 143 const std::unordered_map<string, Tensor>* cached_values, Graph* out_graph, in ExtractConstantSubgraph() argument 323 if (cached_values != nullptr) { in ExtractConstantSubgraph() 324 auto it = cached_values->find(output_tensor_name); in ExtractConstantSubgraph() 325 if (it != cached_values->end() && in ExtractConstantSubgraph() 352 std::unordered_map<string, Tensor>* cached_values, in EvaluateConstantTensor() argument 394 TF_RETURN_IF_ERROR(ExtractConstantSubgraph(*src, refiner, cached_values, in EvaluateConstantTensor() 428 if (cached_values != nullptr && in EvaluateConstantTensor() 430 (*cached_values)[output_tensor_name] = outputs[0]; in EvaluateConstantTensor()
|
D | eval_const_tensor.h | 63 std::unordered_map<string, Tensor>* cached_values = nullptr,
|
/external/tensorflow/tensorflow/python/kernel_tests/boosted_trees/ |
D | prediction_ops_test.py | 1500 cached_values = [[0.08], [0.093], [0.0553], [0.0783], [0.15 + 0.08], 1503 logits_updates + cached_values) 1663 cached_values = np.array([[0.01 + 0.07, 0.032 + 0.02], 1671 np.array(logits_updates) + cached_values) 1813 cached_values = [[0.08], [0.093], [0.0553], [0.0783], [0.15 + 0.08], 1819 logits_updates + cached_values) 2004 cached_values = np.array([[0.01 + 0.07, 0.032 + 0.02], 2011 np.array(logits_updates) + cached_values)
|