D | metal_spatial_tensor.cc | 535 id<MTLTexture> texture_buffer = nullptr; in SetBufferHandle() local 537 CreateTextureBuffer(memory_, shape_, descriptor_, &texture_buffer)); in SetBufferHandle() 538 texture_mem_ = texture_buffer; in SetBufferHandle() 562 id<MTLTexture> texture_buffer = nullptr; in CreateSharedBufferTensor() local 565 CreateTextureBuffer(buffer, shape5D, descriptor, &texture_buffer)); in CreateSharedBufferTensor() 567 *result = MetalSpatialTensor(buffer, texture_buffer, false, true, shape5D, in CreateSharedBufferTensor() 575 id<MTLTexture> texture_buffer = nullptr; in CreateSharedBufferTensor() local 578 CreateTextureBuffer(buffer, shape, descriptor, &texture_buffer)); in CreateSharedBufferTensor() 580 *result = MetalSpatialTensor(buffer, texture_buffer, false, true, shape, in CreateSharedBufferTensor()
|