Searched refs:is_entry_function (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | mlir_hlo_to_hlo.cc | 484 Block* block, xla::XlaBuilder* builder, bool is_entry_function,
504 mlir::Operation* inst, bool is_entry_function,
1209 mlir::Operation* inst, bool is_entry_function, in Lower() argument
1305 if ((return_tuple_ && is_entry_function) || num_return_values != 1) { in Lower()
1317 if (!is_entry_function || !has_ret_shardings) continue; in Lower()
1508 Block* block, xla::XlaBuilder* builder, bool is_entry_function, in LowerBasicBlockAsFunction() argument
1518 if (is_entry_function && use_tuple_args_) { in LowerBasicBlockAsFunction()
1561 if (failed(Lower(&inst, is_entry_function, ret_shardings, builder, in LowerBasicBlockAsFunction()
|