- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for getThenBranch (0.35 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/functional_control_flow_to_regions.cc
builder.getStringAttr(if_op.else_function().getName())); CopyDeviceAndUnderscoredAttributes(if_op, if_region); CreateCall(if_op, if_op.then_function(), /*caller_region=*/if_region.getThenBranch(), if_op.getInput(), /*use_region_args=*/false, /*forward_block_args=*/false); CreateCall(if_op, if_op.else_function(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc
for (const auto& entry : output_buffer_to_size) { (*buffer_to_size)[new_op.getResult(std::get<0>(entry))] = { new_op.getResult(std::get<1>(entry)), std::get<2>(entry)}; } new_op.getThenBranch().takeBody(if_op.getThenBranch()); new_op.getElseBranch().takeBody(if_op.getElseBranch()); if_op.replaceAllUsesWith( new_op.getResults().take_front(if_op.getNumResults())); if_op.erase(); return success(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/decompose_reduce_dataset.cc
} builder.create<TF::YieldOp>(loc, /*operands=*/else_returns); // Then branch gets the data and calls the reduce_function. auto& then_branch = dataset_if.getThenBranch(); then_branch.push_back(new Block); builder.setInsertionPointToEnd(&then_branch.front()); // Add iterator operational data access inside if.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/region_control_flow_to_functional.cc
// collect the call arguments post lookup through cast Op's extern_values.push_back(first); return true; }; const TrivialTransformInfo tti(IsSingleCallRegion(if_region.getThenBranch()), IsSingleCallRegion(if_region.getElseBranch()), if_arg_matcher); std::string then_name, else_name; if (tti.can_transform) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_saved_model_freeze_variables.cc
} } else if (auto if_op = dyn_cast<TF::IfRegionOp>(user_op)) { (*arguments_to_erase)[if_op].push_back(argument_index); for (auto callee : {&if_op.getThenBranch(), &if_op.getElseBranch()}) { work_list->push_back(std::make_pair(callee, argument_index)); } } else if (auto while_op = dyn_cast<TF::WhileOp>(user_op)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 09:56:53 UTC 2024 - 19.4K bytes - Viewed (0)