should_allocate (definition) in projects: external

Searched defs:should_allocate (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
D	cpu_function_runtime.cc	`62 bool should_allocate = in AlignedBufferBytes() local 89 bool should_allocate = in MallocContiguousBuffers() local`
/external/pytorch/torch/_inductor/
D	ir.py	3159 def should_allocate(self): member in Buffer 3505 def should_allocate(self): member in ComputedBuffer 3551 def should_allocate(self): member in TemplateBuffer 3924 def should_allocate(self): member in ConcatKernel 4581 def should_allocate(self): member in ExternKernelOut 4640 def should_allocate(self): member in ExternKernelAlloc 4691 def should_allocate(self): member in UserDefinedTritonKernel 4786 def should_allocate(self): member in MutationOutput 4818 def should_allocate(self): member in InplaceBernoulliFallback 4855 def should_allocate(self): member in InplaceCopyFallback [all …]