Search Options

Results per page
Sort
Preferred Languages
Advance

Results 21 - 30 of 93 for getAxes (0.18 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/replicate_tensor_list_init_ops_pass.cc

    template <typename T>
    void ReplicateTensorListForUses(T tensor_list_op) {
      Value tensor_list = tensor_list_op.getResult();
      std::vector<OpOperand*> uses;
      for (auto& use : tensor_list.getUses()) {
        uses.emplace_back(&use);
      }
      OpBuilder builder(tensor_list_op.getOperation());
      for (OpOperand* operand : uses) {
        auto new_op = builder.clone(*tensor_list_op.getOperation());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sun Jan 22 17:28:34 UTC 2023
    - 2.7K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/constant_op_device_assignment.cc

        }
        OpBuilder builder(op);
        llvm::StringMap<mlir::Operation *> cloned_op_by_device;
        bool all_uses_replaced = true;
    
        for (mlir::OpOperand &use :
             llvm::make_early_inc_range(op.getResult().getUses())) {
          mlir::Operation *user_op = use.getOwner();
          StringAttr device_attr = user_op->getAttrOfType<StringAttr>(kDeviceAttr);
          if (!device_attr) {
            all_uses_replaced = false;
            continue;
          }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Oct 05 23:50:19 UTC 2022
    - 3.3K bytes
    - Viewed (0)
  3. pkg/volume/iscsi/iscsi.go

    	secret, err := createSecretMap(spec, plugin, pod.Namespace)
    	if err != nil {
    		return nil, err
    	}
    	return plugin.newMounterInternal(spec, pod.UID, &ISCSIUtil{}, plugin.host.GetMounter(plugin.GetPluginName()), plugin.host.GetExec(plugin.GetPluginName()), secret)
    }
    
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Tue May 14 06:17:25 UTC 2024
    - 23.4K bytes
    - Viewed (0)
  4. pkg/controller/volume/persistentvolume/volume_host.go

    	return func(types.UID) {
    		//nolint:logcheck
    		klog.ErrorS(nil, "DeleteServiceAccountToken unsupported in PersistentVolumeController")
    	}
    }
    
    func (adc *PersistentVolumeController) GetExec(pluginName string) utilexec.Interface {
    	return utilexec.New()
    }
    
    func (ctrl *PersistentVolumeController) GetNodeLabels() (map[string]string, error) {
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Thu Apr 18 11:00:37 UTC 2024
    - 4.5K bytes
    - Viewed (0)
  5. 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)
  6. 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)
  7. 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)
  8. 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)
  9. 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)
  10. 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)
Back to top