Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dtensor_list_ops_decomposition.cc122 output_buffer_to_size; in AddTensorListSizesToTerminator() local
126 output_buffer_to_size.emplace_back(retval.index(), new_outputs.size(), in AddTensorListSizesToTerminator()
133 return output_buffer_to_size; in AddTensorListSizesToTerminator()
143 auto output_buffer_to_size = AddTensorListSizesToTerminator<func::ReturnOp>( in ModifyFunctionReturn() local
146 return output_buffer_to_size; in ModifyFunctionReturn()
173 auto output_buffer_to_size = ModifyFunctionReturn(body, body_map); in HandleWhileOp() local
185 if (output_buffer_to_size.empty()) { in HandleWhileOp()
198 for (const auto& entry : output_buffer_to_size) { in HandleWhileOp()
240 auto output_buffer_to_size = in HandleCaseOrIfOp() local
245 if (output_buffer_to_size.empty() && arg_no_changed) return success(); in HandleCaseOrIfOp()
[all …]