Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_merge_variables_with_execute.cc88 int execute_output_index = -1; member
263 info.execute_output_index = execute_output.index(); in BuildVariableAccessInfo()
292 output_fused[info_it->second.execute_output_index] = false; in BuildVariableAccessInfo()
293 info_it->second.execute_output_index = -1; in BuildVariableAccessInfo()
463 if (info.execute_input_index < 0 || info.execute_output_index < 0) continue; in UpdateSerializedModule()
471 builder.getI64IntegerAttr(info.execute_output_index)); in UpdateSerializedModule()
475 assert(aliasing_attr.getInt() == info.execute_output_index); in UpdateSerializedModule()
522 device_var_updates_indices.push_back(info.execute_output_index); in MergeForOneTPUExecute()