Search Options

Results per page
Sort
Preferred Languages
Advance

Results 61 - 70 of 78 for getAxes (0.2 sec)

  1. tensorflow/compiler/mlir/tf2xla/internal/passes/extract_outside_compilation.cc

    // `cluster_ops` with a non-static shape.
    bool HasDynamicOutputs(const llvm::SmallSetVector<Operation*, 4>& cluster_ops) {
      for (Operation* op : cluster_ops) {
        for (const OpOperand& use : op->getUses()) {
          if (cluster_ops.count(use.getOwner())) {
            continue;
          }
          if (mlir::TF::CanBeRefined(use.get().getType())) return true;
        }
      }
      return false;
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 21:25:12 UTC 2024
    - 68.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc

        if (!shift_ty || shift_ty.getRank() != 0) {
          return rewriter.notifyMatchFailure(
              op, "require the type of shift to be 0D tensor");
        }
    
        APInt val;
        if (!matchPattern(op.getAxis(), m_ConstantInt(&val))) {
          return rewriter.notifyMatchFailure(op, "require axis to be constant");
        }
        int axis = val.getSExtValue();
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 291.8K bytes
    - Viewed (0)
  3. pkg/volume/util/util.go

    // and Exec taken from given VolumeHost.
    func NewSafeFormatAndMountFromHost(pluginName string, host volume.VolumeHost) *mount.SafeFormatAndMount {
    	mounter := host.GetMounter(pluginName)
    	exec := host.GetExec(pluginName)
    	return &mount.SafeFormatAndMount{Interface: mounter, Exec: exec}
    }
    
    // GetVolumeMode retrieves VolumeMode from pv.
    // If the volume doesn't have PersistentVolume, it's an inline volume,
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Fri May 31 12:32:15 UTC 2024
    - 28.8K bytes
    - Viewed (0)
  4. pkg/controller/volume/attachdetach/attach_detach_controller.go

    	return func(types.UID) {
    		// nolint:logcheck
    		klog.ErrorS(nil, "DeleteServiceAccountToken unsupported in attachDetachController")
    	}
    }
    
    func (adc *attachDetachController) GetExec(pluginName string) utilexec.Interface {
    	return utilexec.New()
    }
    
    func (adc *attachDetachController) addNodeToDswp(node *v1.Node, nodeName types.NodeName) {
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Sat May 04 18:33:12 UTC 2024
    - 32.6K bytes
    - Viewed (0)
  5. pkg/kubelet/container/runtime.go

    // streaming calls (exec/attach/port-forward) themselves. In this case, Kubelet should redirect to
    // the runtime server.
    type StreamingRuntime interface {
    	GetExec(ctx context.Context, id ContainerID, cmd []string, stdin, stdout, stderr, tty bool) (*url.URL, error)
    	GetAttach(ctx context.Context, id ContainerID, stdin, stdout, stderr, tty bool) (*url.URL, error)
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Fri Jun 07 00:05:23 UTC 2024
    - 27.8K bytes
    - Viewed (0)
  6. pkg/kubelet/kubelet_pods.go

    	}
    	// TODO(tallclair): Pass a proper timeout value.
    	return kl.runner.RunInContainer(ctx, container.ID, cmd, 0)
    }
    
    // GetExec gets the URL the exec will be served from, or nil if the Kubelet will serve it.
    func (kl *Kubelet) GetExec(ctx context.Context, podFullName string, podUID types.UID, containerName string, cmd []string, streamOpts remotecommandserver.Options) (*url.URL, error) {
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Fri Jun 14 16:09:17 UTC 2024
    - 101.2K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.h

            }
            TensorRangeSanityCheck(op, rmin, rmax);
            mins.push_back(rmin);
            maxs.push_back(rmax);
          }
          quant_type = quantfork::fakeQuantAttrsToType(
              op.getLoc(), num_bits, *op.getAxis(), mins, maxs, narrow_range,
              expressed, is_signed);
          if (legacy_float_scale) {
            quant_type = DownCastScale(quant_type, mins, maxs, op->getLoc());
          }
        } else if (auto stats =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 20:30:06 UTC 2024
    - 41.7K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_communication.cc

          value.replaceAllUsesWith(new_element.getResult());
        }
        return;
      }
    
      Value sub_tuple;
      for (auto& use : llvm::make_early_inc_range(value.getUses())) {
        if (isa<GetTupleElementOp>(use.getOwner())) {
          use.set(replacement);
          continue;
        }
    
        if (!sub_tuple)
          sub_tuple = CreateSubTuple(builder, replacement, tuple_type.size());
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 40.5K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/flatbuffer_export.cc

      // marked as a stateful. If so, set the tensor's is_variable as true
      // This is v1 ref variable semantics in the TFLite runtime.
      bool is_variable = false;
      for (auto& use : value.getUses()) {
        is_variable = IsStatefulOperand(use.getOwner(), use.getOperandNumber());
        if (is_variable) {
          break;
        }
      }
      // The value is used as a variable if produced by an op with "tfl.is_variable"
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:41:49 UTC 2024
    - 164.5K bytes
    - Viewed (0)
  10. pkg/volume/plugins.go

    	DeleteServiceAccountTokenFunc() func(podUID types.UID)
    
    	// Returns an interface that should be used to execute any utilities in volume plugins
    	GetExec(pluginName string) exec.Interface
    
    	// Returns the labels on the node
    	GetNodeLabels() (map[string]string, error)
    
    	// Returns the name of the node
    	GetNodeName() types.NodeName
    
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Thu Jun 06 16:13:15 UTC 2024
    - 38.2K bytes
    - Viewed (0)
Back to top