Searched refs:variable_reads (Results 1 – 2 of 2) sorted by relevance
342 const std::vector<Node*>& variable_reads, Graph* graph,377 std::vector<Node*>* variable_reads);438 Node* compile_node, const std::vector<Node*>& variable_reads,540 std::vector<Node*>* variable_reads);
2235 const ParameterInfo& params_info, const std::vector<Node*>& variable_reads, in BuildDynamicShapeNodes() argument2277 TF_RET_CHECK(0 <= var_num && var_num < variable_reads.size()); in BuildDynamicShapeNodes()2278 src = variable_reads[var_num]; in BuildDynamicShapeNodes()2583 Graph* graph, std::vector<Node*>* variable_reads) { in BuildVariableReads() argument2584 variable_reads->resize(variables.size()); in BuildVariableReads()2599 (*variable_reads)[i] = read_node = graph->AddNode(def, &status); in BuildVariableReads()2791 const std::vector<Node*>& variable_reads, in CreateOrGetPerHostVariableCopy() argument2813 index_mapping.reserve(variable_reads.size()); in CreateOrGetPerHostVariableCopy()2814 dtypes.reserve(variable_reads.size()); in CreateOrGetPerHostVariableCopy()2815 for (int64 i = 0; i < variable_reads.size(); ++i) { in CreateOrGetPerHostVariableCopy()[all …]