Search Options

Results per page
Sort
Preferred Languages
Advance

Results 21 - 30 of 87 for getAxes (0.28 sec)

  1. tensorflow/compiler/mlir/tensorflow/utils/topological_sort.cc

          Value value = todo.front();
          todo.pop();
          // All operations that have all their inputs available are good to go.
          // Uses, not Users, in case getUsers ever dedups.
          for (OpOperand& operand : value.getUses()) {
            Operation* user = ancestor[operand.getOwner()];
            remaining_incoming_data_edges[user]--;
            if (remaining_incoming_data_edges[user] == 0 &&
                remaining_incoming_ctrl_edges[user] == 0) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Nov 08 17:01:11 UTC 2022
    - 5K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/tpu_parallel_execute_sink_resource_write.cc

    // before or above the parallel_execute.
    TF::AssignVariableOp GetSingleUseResourceWrite(
        tf_device::ParallelExecuteOp parallel_execute, Value result) {
      if (!result.hasOneUse()) return nullptr;
    
      OpOperand& use = *result.getUses().begin();
      auto assign_var = dyn_cast<TF::AssignVariableOp>(use.getOwner());
      if (!assign_var) return nullptr;
    
      if (use.get() != assign_var.getValue()) return nullptr;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Dec 06 04:46:18 UTC 2022
    - 6.6K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/quantization/ir/ConvertSimQuant.cc

        for (auto m : fqOp.getMax())
          max.push_back(cast<FloatAttr>(m).getValueAsDouble());
    
        return fakeQuantAttrsToType(fqOp.getLoc(), fqOp.getNumBits(),
                                    fqOp.getAxis(), min, max, fqOp.getNarrowRange(),
                                    expressedType, fqOp.getIsSigned());
      }
    };
    
    void ConvertSimulatedQuantPass::runOnOperation() {
      bool hadFailure = false;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 08 02:10:16 UTC 2024
    - 6K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tf2xla/internal/passes/hoist_broadcast_read.cc

        Value res = read.getResource();
        Operation* scope = res.getParentBlock()->getParentOp();
        if (!scope->isProperAncestor(replicate)) continue;
        bool has_conflicting_write = false;
        for (OpOperand& use : res.getUses()) {
          Operation* using_op = use.getOwner();
          if (using_op == read) continue;
          if (!replicate->isProperAncestor(using_op)) continue;
          Operation* peer = GetAncestorBelow(using_op, replicate);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 5.3K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/fused_kernel_matcher.cc

    // multiple such activations, one is returned (with no guarantee as to which
    // one). If there are no activation functions that use the output, returns
    // nullptr.
    Operation *GetActivation(Value op) {
      for (auto &use : op.getUses()) {
        if (IsActivationFunction(use.getOwner())) return use.getOwner();
      }
      return nullptr;
    }
    
    // Finds and returns a BiasAdd that uses the result of `op` as the `value`
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 14.9K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/quantization/tensorflow/passes/propagate_quantize_type.cc

        llvm::SmallVector<Operation*> users(op->getUsers().begin(),
                                            op->getUsers().end());
    
        bool changed = false;
        for (auto& use : op->getUses()) {
          Operation* user_op = use.getOwner();
          int user_idx = use.getOperandNumber();
          if (!IsOpWithInt8TypeOperand(user_op)) continue;
          // If the next op is terminator, function type needs to be changed so
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 7K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/lite/stablehlo/transforms/fuse_convolution_pass.cc

          });
        }
        // For dynamic case, the result of conv should be used by shape_of and mul.
        if (is_dynamic_broadcast) {
          auto conv_uses = (*conv_op.getODSResults(0).begin()).getUses();
          if (std::distance(conv_uses.begin(), conv_uses.end()) != 2 ||
              quant::FindUserOfType<shape::ShapeOfOp>(conv_op) == nullptr ||
              quant::FindUserOfType<mhlo::MulOp>(conv_op) == nullptr) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 22:21:19 UTC 2024
    - 8.3K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/transforms/cluster_formation.cc

      for (const auto& p : llvm::zip(live_outs, launch_op.getResults())) {
        Value from = std::get<0>(p);
        // TODO(jingpu): move this to RegionUtils.h in MLIR core.
        for (auto& use : llvm::make_early_inc_range(from.getUses())) {
          if (launch_op_region->isAncestor(use.getOwner()->getParentRegion()))
            continue;
          use.set(std::get<1>(p));
        }
      }
    }
    
    // Get all escaped live-out values of a region.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Oct 05 13:30:21 UTC 2023
    - 6.8K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/transforms/legalize_hashtables.cc

               mlir::isa<IntegerType>(key_dtype) &&
               mlir::cast<IntegerType>(key_dtype).getWidth() == 64))) {
          return false;
        }
    
        for (auto& use : hashtable->getUses()) {
          Operation* user = use.getOwner();
    
          // Allow consuming hash table ops that can be covered by TensorFlow Lite
          // hash table kernels.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 7.6K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/transforms/group_by_dialect.cc

      }
    
      // All results in our set that have a user outside our set.
      llvm::DenseSet<Value> outputs_seen;
      for (Operation* op : ops) {
        for (Value result : op->getResults()) {
          for (auto& use : result.getUses()) {
            if (!all_operations.contains(use.getOwner())) {
              if (!outputs_seen.contains(result)) {
                outputs->push_back(result);
                outputs_seen.insert(result);
              }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 17 07:31:01 UTC 2023
    - 8K bytes
    - Viewed (0)
Back to top