Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 1,341 for Devices (0.24 sec)

  1. pkg/kubelet/apis/podresources/server_v1_test.go

    							{
    								Name:             containerName,
    								Devices:          []*podresourcesapi.ContainerDevices{},
    								DynamicResources: []*podresourcesapi.DynamicResource{},
    							},
    						},
    					},
    				},
    			},
    		},
    		{
    			desc:             "pod with devices",
    			pods:             pods,
    			devices:          devs,
    			cpus:             cpus,
    			memory:           memory,
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Wed Apr 24 18:25:29 UTC 2024
    - 45.9K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/device_util.h

    };
    
    // Caches some miscellaneous information about TF devices.  Thread compatible.
    class DeviceInfoCache {
     public:
      bool IsGpu(DeviceId device) const { return is_gpu_[device.id()]; }
      bool IsCpu(DeviceId device) const { return is_cpu_[device.id()]; }
    
      absl::string_view GetNameFor(DeviceId device) const {
        return names_[device.id()];
      }
    
      absl::StatusOr<DeviceId> GetIdFor(absl::string_view name);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 15 17:18:31 UTC 2024
    - 7.1K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/tpu_device_propagation.cc

        value_to_device.insert({result, device});
      }
    }
    
    // Checks if an operation can have TPU devices propagated through.
    bool IsSupportedOpToSetDevice(Operation& op) {
      return IsSupportedExecutorOp(op) ||
             isa<TF::IdentityOp, TF::IdentityNOp, TF::ShapeOp>(op);
    }
    
    // Finds nonconflicting TPU device for an operation from its operands. If an
    // operand has no device or a non TPU device, or if there are conflicting
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 11K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/tests/tpu_rewrite.mlir

    }
    
    // -----
    
    // Tests collecting compilation and execution devices results in an error.
    
    module attributes {tf.versions = {producer = 888 : i32}, tf.devices = ["/job:worker/replica:0/task:0/device:CPU:0", "/job:worker/replica:0/task:0/device:TPU:0"]} {
      func.func @bad_devices() {
        // expected-error@+1 {{error in fetching TPU compilation/execution devices: no TPU_SYSTEM devices found}}
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 22:03:30 UTC 2024
    - 172.9K bytes
    - Viewed (0)
  5. pkg/kubelet/cm/dra/claiminfo_test.go

    							"vendor.com/device=device1",
    							"vendor.com/device=device2",
    						},
    						"test-plugin2": {
    							"vendor.com/device=device1",
    							"vendor.com/device=device2",
    						},
    					},
    				},
    			},
    			expectedResult: []kubecontainer.CDIDevice{
    				{
    					Name: "vendor.com/device=device1",
    				},
    				{
    					Name: "vendor.com/device=device1",
    				},
    				{
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Fri May 03 13:30:31 UTC 2024
    - 21K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_variable_runtime_reformatting.cc

        device_list_for_alias.reserve(device_list.size());
    
        for (auto device : device_list)
          device_list_for_alias.emplace_back(
              mlir::cast<StringAttr>(device).getValue());
    
        devices.insert({device_alias, device_list_for_alias});
      }
    
      OpBuilder builder(replicate);
      builder.setInsertionPoint(while_op);
      // Create per-device variables for formatting state, and add them to the while
      // loop.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 21.9K bytes
    - Viewed (0)
  7. tensorflow/compiler/jit/xla_host_send_recv_device_context_test.cc

        SessionOptions options;
        std::vector<std::unique_ptr<Device>> devices;
        Status s = device_factory->CreateDevices(
            options, "/job:worker/replica:0/task:0", &devices);
        device_ = std::move(devices[0]);
    
        AllocatorAttributes host_alloc_attr;
        host_alloc_attr.set_on_host(true);
        host_allocator_ = device_->GetAllocator(host_alloc_attr);
    
        AllocatorAttributes device_alloc_attr;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 22:46:36 UTC 2024
    - 7.2K bytes
    - Viewed (0)
  8. pkg/kubelet/cm/dra/claiminfo.go

    func (info *ClaimInfo) cdiDevicesAsList() []kubecontainer.CDIDevice {
    	var cdiDevices []kubecontainer.CDIDevice
    	for _, devices := range info.CDIDevices {
    		for _, device := range devices {
    			cdiDevices = append(cdiDevices, kubecontainer.CDIDevice{Name: device})
    		}
    	}
    	return cdiDevices
    }
    
    // addPodReference adds a pod reference to the claim info.
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Fri May 03 13:30:31 UTC 2024
    - 7.9K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/ir/tf_device.cc

        Builder* builder, OperationState* state, int n,
        std::optional<DictionaryAttr> devices,
        llvm::ArrayRef<std::pair<ValueRange, Type>> replicated_inputs,
        ValueRange packed_inputs, TypeRange replica_output_types) {
      DCHECK_GE(n, 2);
      state->addAttribute("n", builder->getI32IntegerAttr(n));
    
      if (devices.has_value()) state->addAttribute("devices", devices.value());
    
      Region* region = state->addRegion();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 33.4K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_rewrite_pass.cc

        // TODO(jpienaar): Remove this later.
        if (auto device = res->getAttrOfType<StringAttr>("device")) {
          if (!device.getValue().empty())
            result_id->setAttr("device", device);
          else
            result_id->setAttr("device", compile_device_op);
        } else if (compile_device_op) {
          result_id->setAttr("device", compile_device_op);
        }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 21:25:12 UTC 2024
    - 29.7K bytes
    - Viewed (0)
Back to top