Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dresource_op_lifting.cc1166 CallOpType call_op, PartitionedCallLiftingInfo& lifting_info) { in UpdatePartitionedCallOpWithNewCallee() argument
1167 if (!lifting_info.lifted_callee) return; in UpdatePartitionedCallOpWithNewCallee()
1170 for (const auto& entry : lifting_info.old_outputs_aliasing_old_inputs) { in UpdatePartitionedCallOpWithNewCallee()
1179 FilterRange<Value, OperandRange>(call_op.args(), lifting_info.use_info); in UpdatePartitionedCallOpWithNewCallee()
1181 call_op.getLoc(), lifting_info.lifted_callee.getType().getResults(), in UpdatePartitionedCallOpWithNewCallee()
1184 "f", builder.getSymbolRefAttr(lifting_info.lifted_callee.getName())); in UpdatePartitionedCallOpWithNewCallee()
1186 new_call, lifting_info.arg_data_type_and_updated_output_index); in UpdatePartitionedCallOpWithNewCallee()
1188 for (int64_t i = 0, end = lifting_info.old_to_new_output_indices.size(); in UpdatePartitionedCallOpWithNewCallee()
1190 if (lifting_info.old_to_new_output_indices[i] >= 0) { in UpdatePartitionedCallOpWithNewCallee()
1192 new_call.getResult(lifting_info.old_to_new_output_indices[i])); in UpdatePartitionedCallOpWithNewCallee()