Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 8 of 8 for GatherNd (0.18 sec)

  1. tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td

        TF_Int64Tensor:$iter_id
      );
    }
    
    def TF_ResourceGatherNdOp : TF_Op<"ResourceGatherNd", []> {
      let summary = "GatherNd on a resource.";
    
      let description = [{
    This op reads the variable referenced by the first argument, and
    then performs a GatherNd operation on it.
      }];
    
      let arguments = (ins
        Arg<TF_ResourceTensor, "", [TF_VariableRead]>:$resource,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 04:08:35 UTC 2024
    - 90.5K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/tests/prepare-tf.mlir

      %0 = "tf.TensorScatterAdd"(%arg0, %arg1, %arg2) : (tensor<7xi64>, tensor<1x1xi32>, tensor<1xi64>) -> tensor<7xi64>
      func.return %0 : tensor<7xi64>
    
      // CHECK-LABEL: scatter_nd_add
      // CHECK:  %[[GATHER:.*]] = "tf.GatherNd"(%arg0, %arg1) <{bad_indices_policy = ""}> : (tensor<7xi64>, tensor<1x1xi32>) -> tensor<1xi64>
      // CHECK:  %[[ADD:.*]] = "tf.Add"(%arg2, %[[GATHER]]) : (tensor<1xi64>, tensor<1xi64>) -> tensor<1xi64>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 29 07:26:59 UTC 2024
    - 59.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/mark_for_compilation_pass.cc

    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 12:19:41 UTC 2024
    - 85.3K bytes
    - Viewed (0)
  4. tensorflow/cc/gradients/math_grad.cc

                               Output& zero_clipped_indices, Output& is_positive) {
      auto gathered = Gather(scope, params, zero_clipped_indices);
      // Replace gathered params of negative indices with 0.
      auto zero_slice = ZerosLike(scope, gathered);
      return SelectV2(scope, is_positive, gathered, zero_slice);
    }
    
    Status UnsortedSegmentMinOrMaxGrad(const Scope& scope, const Operation& op,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Aug 25 18:20:20 UTC 2023
    - 50.7K bytes
    - Viewed (0)
  5. pkg/proxy/winkernel/proxier.go

    	endpointUpdateResult := proxier.endpointsMap.Update(proxier.endpointsChanges)
    
    	deletedUDPClusterIPs := serviceUpdateResult.DeletedUDPClusterIPs
    	// merge stale services gathered from EndpointsMap.Update
    	for _, svcPortName := range endpointUpdateResult.NewlyActiveUDPServices {
    		if svcInfo, ok := proxier.svcPortMap[svcPortName]; ok && svcInfo != nil && svcInfo.Protocol() == v1.ProtocolUDP {
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Thu Apr 18 13:25:06 UTC 2024
    - 59K bytes
    - Viewed (0)
  6. pkg/controller/nodelifecycle/node_lifecycle_controller.go

    		nc.podUpdateQueue.AddRateLimited(podItem)
    		return
    	}
    
    	nodeName := pod.Spec.NodeName
    
    	nodeHealth := nc.nodeHealthMap.getDeepCopy(nodeName)
    	if nodeHealth == nil {
    		// Node data is not gathered yet or node has been removed in the meantime.
    		return
    	}
    
    	_, err = nc.nodeLister.Get(nodeName)
    	if err != nil {
    		logger.Info("Failed to read node", "node", klog.KRef("", nodeName), "err", err)
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Sat May 04 18:33:12 UTC 2024
    - 51.6K bytes
    - Viewed (0)
  7. staging/src/k8s.io/apiserver/pkg/server/filters/priority-and-fairness_test.go

    				continue
    			}
    
    			labelMatch = true
    			sum += int(metric.GetGauge().GetValue())
    		}
    	}
    	if !familyMatch {
    		return fmt.Errorf("expected to find the metric family: %s in the gathered result", name)
    	}
    	if !labelMatch {
    		return fmt.Errorf("expected to find metrics with matching labels: %#+v", labelFilter)
    	}
    	if wantValue != sum {
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Mon Oct 30 12:18:40 UTC 2023
    - 52.6K bytes
    - Viewed (0)
  8. src/cmd/go/internal/test/test.go

    		// See runtime/unsafepoint_test.go.
    		pmain.Internal.OmitDebug = false
    	}
    
    	if !cfg.BuildN {
    		// writeTestmain writes _testmain.go,
    		// using the test description gathered in t.
    		if err := os.WriteFile(testDir+"_testmain.go", *pmain.Internal.TestmainGo, 0666); err != nil {
    			return nil, nil, nil, err
    		}
    	}
    
    	// Set compile objdir to testDir we've already created,
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Thu May 16 14:34:32 UTC 2024
    - 71.9K bytes
    - Viewed (0)
Back to top