Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for set_device (0.18 sec)

  1. tensorflow/c/kernels_test.cc

                                                   const char* node_name,
                                                   Status* status) {
      NodeDef def;
      def.set_op(op_name);
      def.set_name(node_name);
      def.set_device(device_name);
      def.add_input("input1");
      def.add_input("input2");
    
      AttrValue v;
      v.set_type(DataType::DT_FLOAT);
      (*def.mutable_attr())["SomeDataTypeAttr"] = v;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 50.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/encapsulate_subgraphs_pass.cc

      // name_in is copied here because name may be modified below if
      // rewrite_subgraph_fn is true.
      string name = name_in;
      call_node_def_.set_op(name);
      call_node_def_.set_name(name);
      call_node_def_.set_device(device_);
    
      if (rewrite_subgraph_fn) {
        std::vector<OutputTensor> arg_source_tensors(args_by_src_.size());
        for (const auto& arg : args_by_src_) {
          arg_source_tensors.at(arg.second) = arg.first;
        }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 51K bytes
    - Viewed (0)
  3. pkg/kubelet/volumemanager/reconciler/reconciler_test.go

    	assert.NoError(t, volumetesting.VerifyZeroDetachCallCount(fakePlugin))
    }
    
    // Populates desiredStateOfWorld cache with one volume/pod.
    // Calls Run()
    // Verifies there are attach/get map paths/setupDevice calls and
    // no detach/teardownDevice calls.
    func Test_Run_Positive_VolumeAttachAndMap(t *testing.T) {
    	pod := &v1.Pod{
    		ObjectMeta: metav1.ObjectMeta{
    			Name:      "pod1",
    			UID:       "pod1uid",
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Tue Apr 09 07:34:33 UTC 2024
    - 75.4K bytes
    - Viewed (0)
  4. pkg/volume/testing/testing.go

    	fv.RLock()
    	defer fv.RUnlock()
    	return fv.TearDownCallCount
    }
    
    func (fv *FakeVolume) TearDownAt(dir string) error {
    	return os.RemoveAll(dir)
    }
    
    // Block volume support
    func (fv *FakeVolume) SetUpDevice() (string, error) {
    	fv.Lock()
    	defer fv.Unlock()
    	if fv.VolName == TimeoutOnMountDeviceVolumeName {
    		fv.DeviceMountState[fv.VolName] = deviceMountUncertain
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Fri May 31 12:32:15 UTC 2024
    - 53.3K bytes
    - Viewed (0)
Back to top