Home
last modified time | relevance | path

Searched refs:MetalSpatialTensor (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
H A Dmetal_spatial_tensor.h32 class MetalSpatialTensor : public GPUObject, public GpuSpatialTensor {
34 MetalSpatialTensor() in MetalSpatialTensor() function
39 MetalSpatialTensor(id<MTLBuffer> buffer, id<MTLTexture> texture,
44 MetalSpatialTensor(MetalSpatialTensor&& tensor);
45 MetalSpatialTensor& operator=(MetalSpatialTensor&& tensor);
46 MetalSpatialTensor(const MetalSpatialTensor&) = delete;
47 MetalSpatialTensor& operator=(const MetalSpatialTensor&) = delete;
49 ~MetalSpatialTensor() override { Release(); } in ~MetalSpatialTensor()
84 MetalSpatialTensor* result, uint64_t buffer_offset);
88 int row_bytes_alignment, MetalSpatialTensor* result,
[all …]
H A Dmetal_spatial_tensor.cc157 MetalSpatialTensor::MetalSpatialTensor(id<MTLBuffer> buffer, in MetalSpatialTensor() function in tflite::gpu::metal::MetalSpatialTensor
168 MetalSpatialTensor::MetalSpatialTensor(MetalSpatialTensor&& tensor) in MetalSpatialTensor() function in tflite::gpu::metal::MetalSpatialTensor
179 MetalSpatialTensor& MetalSpatialTensor::operator=(MetalSpatialTensor&& tensor) { in operator =()
193 void MetalSpatialTensor::Release() { in Release()
202 absl::Status MetalSpatialTensor::GetGPUResources( in GetGPUResources()
248 absl::Status MetalSpatialTensor::CreateFromDescriptor( in CreateFromDescriptor()
260 absl::Status MetalSpatialTensor::UploadDescriptorData( in UploadDescriptorData()
265 absl::Status MetalSpatialTensor::ToDescriptor(TensorDescriptor* desc, in ToDescriptor()
274 absl::Status MetalSpatialTensor::WriteData(id<MTLDevice> device, in WriteData()
299 absl::Status MetalSpatialTensor::ReadData(id<MTLDevice> device, in ReadData()
[all …]
H A Dinference_context.h129 MetalSpatialTensor* tensor_ptr);
131 MetalSpatialTensor* GetTensor(ValueId tensor_id);
173 absl::flat_hash_map<ValueId, MetalSpatialTensor*> external_immutable_tensors_;
174 absl::flat_hash_map<ValueId, MetalSpatialTensor*> external_mutable_tensors_;
177 std::map<ValueId, MetalSpatialTensor> const_tensors_;
181 std::vector<MetalSpatialTensor>
185 std::map<ValueId, MetalSpatialTensor> strong_shape_tensors_;
H A Dinference_context.cc157 dynamic_cast<MetalSpatialTensor*>(external_tensor.second); in InitFromGraph()
163 std::map<ValueId, MetalSpatialTensor> temp_external_tensors; in InitFromGraph()
228 std::map<ValueId, MetalSpatialTensor> temp_external_tensors; in RestoreDeserialized()
233 dynamic_cast<MetalSpatialTensor*>(external_tensor.second); in RestoreDeserialized()
324 MetalSpatialTensor* InferenceContext::GetTensor(ValueId tensor_id) { in GetTensor()
347 MetalSpatialTensor* gpu_tensor = GetTensor(id); in SetInputTensor()
355 const MetalSpatialTensor* gpu_tensor = GetTensor(id); in GetOutputTensor()
772 MetalSpatialTensor* tensor_ptr) { in SetTensor()
H A Dcompute_task.cc254 dynamic_cast<const MetalSpatialTensor*>(operation_->GetSrcTensors()[i]); in UpdateParams()
263 dynamic_cast<const MetalSpatialTensor*>(operation_->GetDstTensors()[i]); in UpdateParams()
328 void ComputeTask::SetSrcTensor(MetalSpatialTensor* tensor, int index) { in SetSrcTensor()
334 void ComputeTask::SetDstTensor(MetalSpatialTensor* tensor, int index) { in SetDstTensor()
H A Dcompute_task.h70 void SetSrcTensor(MetalSpatialTensor* tensor, int index);
72 void SetDstTensor(MetalSpatialTensor* tensor, int index);
H A Dmetal_spatial_tensor_test.mm76 tflite::gpu::metal::MetalSpatialTensor tensor;
157 tflite::gpu::metal::MetalSpatialTensor tensor;
H A Dmetal_arguments.cc76 MetalSpatialTensor gpu_tensor; in CreateMetalObject()
78 *result = std::make_unique<MetalSpatialTensor>(std::move(gpu_tensor)); in CreateMetalObject()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/metal/kernels/
H A Dtest_util.cc61 std::vector<MetalSpatialTensor> src(src_cpu.size()); in ExecuteGpuOperationInternal()
67 std::vector<MetalSpatialTensor> dst(dst_cpu.size()); in ExecuteGpuOperationInternal()
143 std::vector<MetalSpatialTensor> src(src_cpu.size()); in ExecuteGPUOperation()
155 std::vector<MetalSpatialTensor> dst(dst_cpu.size()); in ExecuteGPUOperation()
H A Dconv_test.mm322 MetalSpatialTensor src_gpu, dst_gpu;
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/
H A Dmetal_delegate.mm406 MetalSpatialTensor metal_tensor;
411 in_out_tensors_[input] = std::make_unique<MetalSpatialTensor>(std::move(metal_tensor));
438 MetalSpatialTensor metal_tensor;
443 in_out_tensors_[output] = std::make_unique<MetalSpatialTensor>(std::move(metal_tensor));
627 std::map<ValueId, std::unique_ptr<MetalSpatialTensor>> in_out_tensors_;