- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 23 for genAllocator (0.19 sec)
-
src/cmd/compile/internal/ssa/_gen/allocators.go
fmt.Fprintln(w, "\"internal/unsafeheader\"") fmt.Fprintln(w, "\"math/bits\"") fmt.Fprintln(w, "\"sync\"") fmt.Fprintln(w, "\"unsafe\"") fmt.Fprintln(w, ")") for _, a := range allocators { genAllocator(w, a) } for _, d := range deriveds { for _, base := range allocators { if base.name == d.base { genDerived(w, d, base) break } } } // gofmt result b := w.Bytes()
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Fri Nov 17 23:34:11 UTC 2023 - 6.7K bytes - Viewed (0) -
tensorflow/compiler/jit/device_context_test.cc
tensorflow::AllocatorAttributes host_alloc_attr; host_alloc_attr.set_on_host(true); host_allocator_ = device_->GetAllocator(host_alloc_attr); tensorflow::AllocatorAttributes device_alloc_attr; device_alloc_attr.set_on_host(false); device_allocator_ = device_->GetAllocator(device_alloc_attr); tensorflow::DeviceContext* device_context; auto status = device_->TryGetDeviceContext(&device_context);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 3.7K bytes - Viewed (0) -
pkg/registry/core/service/ipallocator/cidrallocator_test.go
// wait for the cidr to be processed and set the informer synced err = wait.PollUntilContextTimeout(context.Background(), 100*time.Millisecond, 5*time.Second, true, func(ctx context.Context) (bool, error) { allocator, err := r.getAllocator(netutils.ParseIPSloppy("192.168.0.1")) if err != nil { t.Logf("unexpected error %v", err) return false, nil } allocator.ipAddressSynced = func() bool { return true }
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Tue Oct 31 21:05:06 UTC 2023 - 13.7K bytes - Viewed (0) -
pkg/registry/core/service/ipallocator/cidrallocator.go
} } func (c *MetaAllocator) AllocateService(service *api.Service, ip net.IP) error { allocator, err := c.getAllocator(ip) if err != nil { return err } return allocator.AllocateService(service, ip) } func (c *MetaAllocator) Allocate(ip net.IP) error { allocator, err := c.getAllocator(ip) if err != nil { return err } return allocator.Allocate(ip) }
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Sat May 04 18:33:12 UTC 2024 - 13.2K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_platform_info.cc
custom_allocator); } std::shared_ptr<se::DeviceMemoryAllocator> GetAllocator( DeviceBase* device, se::Stream* stream, const XlaPlatformInfo& platform_info) { if (platform_info.custom_allocator()) { return platform_info.custom_allocator(); } auto* alloc = device->GetAllocator({}); if (!stream) { // Stream is not set for the host platform. se::Platform* platform =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 17:23:27 UTC 2024 - 17.4K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_host_send_recv_device_context_test.cc
AllocatorAttributes host_alloc_attr; host_alloc_attr.set_on_host(true); host_allocator_ = device_->GetAllocator(host_alloc_attr); AllocatorAttributes device_alloc_attr; device_alloc_attr.set_on_host(false); device_allocator_ = device_->GetAllocator(device_alloc_attr); } protected: std::unique_ptr<Device> device_; Allocator* host_allocator_; Allocator* device_allocator_;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_kernel_creator.cc
&fbody->record->fdef().signature(), node_def, fbody->arg_types, fbody->ret_types); OpKernelConstruction construction(DeviceType(dev->device_type()), dev, dev->GetAllocator(AllocatorAttributes()), flr, dev->resource_manager(), props, input_memory_types, output_memory_types,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 22:24:01 UTC 2023 - 4.4K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_launch_util_gpu_test.cc
// Get the host allocator. AllocatorAttributes host_alloc_attr; host_alloc_attr.set_on_host(true); host_allocator_ = device_->GetAllocator(host_alloc_attr); AllocatorAttributes device_alloc_attr; device_alloc_attr.set_on_host(false); device_allocator_ = device_->GetAllocator(device_alloc_attr); // Create the DeviceCompiler to help with compiling executables.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 10K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_compiler_options_util.cc
options.graph_def_version = function_library.graph_def_version(); options.allow_cpu_custom_calls = (platform_info.platform_id() == se::host::kHostPlatformId); options.device_allocator = GetAllocator(device, stream, platform_info); if (platform_info.xla_device_metadata()) { options.shape_determination_fns = platform_info.xla_device_metadata()->default_shape_determination_fns(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 6.4K bytes - Viewed (0) -
tensorflow/c/kernels/bitcast_op_test.cc
#include "tensorflow/core/platform/test.h" namespace tensorflow { namespace { class DummyDevice : public DeviceBase { public: explicit DummyDevice(Env* env) : DeviceBase(env) {} Allocator* GetAllocator(AllocatorAttributes /*attr*/) override { return cpu_allocator(); } }; void TestBitcastOp(Tensor* input_tensor, DataType out_type, TensorShape expected_shape, error::Code expected_code) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jul 18 15:10:51 UTC 2022 - 5.5K bytes - Viewed (0)