Home
last modified time | relevance | path

Searched refs:cached_values (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Deval_const_tensor.cc128 const std::unordered_map<string, Tensor>* cached_values, Graph* out_graph, in ExtractConstantSubgraph() argument
283 if (cached_values != nullptr) { in ExtractConstantSubgraph()
284 auto it = cached_values->find(output_tensor_name); in ExtractConstantSubgraph()
285 if (it != cached_values->end() && in ExtractConstantSubgraph()
312 std::unordered_map<string, Tensor>* cached_values, in EvaluateConstantTensor() argument
337 TF_RETURN_IF_ERROR(ExtractConstantSubgraph(*src, refiner, cached_values, in EvaluateConstantTensor()
371 if (cached_values != nullptr && in EvaluateConstantTensor()
373 (*cached_values)[output_tensor_name] = outputs[0]; in EvaluateConstantTensor()
Deval_const_tensor.h60 std::unordered_map<string, Tensor>* cached_values = nullptr,
/external/tensorflow/tensorflow/python/kernel_tests/boosted_trees/
Dprediction_ops_test.py652 cached_values = [[0.08], [0.093], [0.0553], [0.0783], [0.15 + 0.08],
655 logits_updates + cached_values)
796 cached_values = [[0.08], [0.093], [0.0553], [0.0783], [0.15 + 0.08],
801 logits_updates + cached_values)