- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for getRegionNumber (0.34 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/test_resource_alias_analysis.cc
for (auto region_arg : TF::filter_resources(region.getArguments())) { InFlightDiagnostic diag = op->emitRemark("Region #") << region.getRegionNumber() << ", Arg #" << region_arg.getArgNumber(); print_aliases(diag, region_arg); } } for (auto result : TF::filter_resources(op->getResults())) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 19 00:13:50 UTC 2022 - 4.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_merge_variables_with_execute.cc
false); auto execute_outputs = parallel_execute ? parallel_execute.GetRegionOutputs( execute_launch->getParentRegion()->getRegionNumber()) : execute_launch.getResults(); for (auto execute_output : llvm::enumerate(execute_outputs)) { // TODO(lyandy): Handle updates to resource writes by remapping to parent
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 29 17:52:11 UTC 2024 - 27K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/mark_input_output_aliases.cc
int operand_idx = use->getOperandNumber(); auto execute_results = execute.GetRegionOutputs( device_return->getParentRegion()->getRegionNumber()); if (operand_idx >= execute_results.size()) return nullptr; auto result_from_use = execute_results[operand_idx]; if (!result_from_use.hasOneUse()) return nullptr;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 04:14:26 UTC 2024 - 7.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/resource_device_inference.cc
LLVM_DEBUG(llvm::dbgs() << " Propagating device = '" << *device << "' to arg #" << output_index << " of region #" << region->getRegionNumber() << "\n"); if (failed(AddResourceDeviceAndEmitError(arg, *device, while_region, result))) return WalkResult::interrupt(); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 03 03:47:00 UTC 2023 - 13.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting.cc
if (write) { info.is_written = true; info.RefineType(write.getValue().getType()); info.write_attrs = user->getAttrDictionary(); written_regions.set(user->getParentRegion()->getRegionNumber()); } } // If the resource is used in an op that we do not understand, skip // lifting for that resource. if (unsupported_use) continue;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 55.1K bytes - Viewed (0)