Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 9 of 9 for getAxes (0.12 sec)

  1. tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc

      // updated in the function.
      llvm::SmallSet<int, 4> indexes;
      for (BlockArgument &arg : func.getArguments()) {
        if (tensor_list_args.contains(arg.getArgNumber())) {
          for (const mlir::OpOperand &use : arg.getUses()) {
            mlir::Operation *op = use.getOwner();
            // Currently we only check if the tensorlist argument is consumed by
            // `TensorListPushBack` or `TensorListResize`, since those are the only
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 70.7K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.cc

            loc, DenseElementsAttr::get(tensorflow::GetTypeFromTFTensorShape(
                                            {}, rewriter.getIntegerType(64)),
                                        op.getAxis()));
        int64_t axis = op.getAxis();
    
        Type prev_input_ty, inferred_ty;
        SmallVector<Value, 4> expanded_inputs;
        expanded_inputs.reserve(op.getN());
        for (Value input : op.getValues()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 74.9K bytes
    - Viewed (0)
  3. pkg/kubelet/server/server_test.go

    	}
    	return s, nil
    }
    
    func (fk *fakeKubelet) GetExec(_ context.Context, podFullName string, podUID types.UID, containerName string, cmd []string, streamOpts remotecommandserver.Options) (*url.URL, error) {
    	if fk.getExecCheck != nil {
    		fk.getExecCheck(podFullName, podUID, containerName, cmd, streamOpts)
    	}
    	// Always use testContainerID
    	resp, err := fk.streamingRuntime.GetExec(&runtimeapi.ExecRequest{
    		ContainerId: testContainerID,
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Wed Apr 24 18:25:29 UTC 2024
    - 51.5K bytes
    - Viewed (0)
  4. pkg/kubelet/kuberuntime/kuberuntime_container.go

    		return remote.ErrContainerStatusNil
    	}
    	return m.ReadLogs(ctx, status.GetLogPath(), containerID.ID, logOptions, stdout, stderr)
    }
    
    // GetExec gets the endpoint the runtime will serve the exec request from.
    func (m *kubeGenericRuntimeManager) GetExec(ctx context.Context, id kubecontainer.ContainerID, cmd []string, stdin, stdout, stderr, tty bool) (*url.URL, error) {
    	req := &runtimeapi.ExecRequest{
    		ContainerId: id.ID,
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Tue Jun 04 06:25:43 UTC 2024
    - 54.7K bytes
    - Viewed (0)
  5. 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)
  6. tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting.cc

        ResourceInfo info;
        info.data_type = GetResourceSubtype(resource);
        llvm::BitVector written_regions(op_->getNumRegions());
        bool unsupported_use = false;
        for (OpOperand& use : resource.getUses()) {
          Operation* user = use.getOwner();
          // If the user is not in one of the regions, we are not interested in it.
          // Since all the sub-regions within this region (i.e., regions attached to
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 55.1K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/lite/flatbuffer_import.cc

            !(qtype.isa<mlir::quant::UniformQuantizedType>() ||
              qtype.isa<mlir::quant::UniformQuantizedPerAxisType>())) {
          return;
        }
        for (auto& use : value.getUses()) {
          Operation* user = use.getOwner();
          if (user->hasTrait<mlir::OpTrait::IsTerminator>()) continue;
    
          auto affine_user = llvm::dyn_cast<mlir::AffineQuantizedOpInterface>(user);
          if (affine_user &&
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 18:21:50 UTC 2024
    - 66.8K bytes
    - Viewed (0)
  8. src/main/java/jcifs/smb/SmbFile.java

            try ( SmbTreeHandleImpl th = ensureTreeConnected() ) {
                SecurityDescriptor desc = querySecurity(th, SecurityInfo.DACL_SECURITY_INFO);
                ACE[] aces = desc.getAces();
                if ( aces != null ) {
                    processAces(aces, resolveSids);
                }
    
                return aces;
            }
        }
    
    
        @Override
    Registered: Wed Jun 12 15:45:55 UTC 2024
    - Last Modified: Thu May 23 01:50:13 UTC 2024
    - 82.3K bytes
    - Viewed (1)
  9. tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_pipelining.cc

        // replication region.
        if (!IsTPUOp(op)) continue;
        for (Value result : op->getResults()) {
          std::vector<std::pair<Operation*, int64_t>> out_of_region_use;
          for (OpOperand& use : result.getUses()) {
            auto use_owner = use.getOwner();
            // TODO(bfontain): Error check here, if the use.getOwner() is not a TPU
            // then this op must be a TPUReplicatedOutputOp.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 92.9K bytes
    - Viewed (0)
Back to top