Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for getRegionNumber (0.34 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
Back to top