Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dmetal_spatial_tensor.h31 class MetalSpatialTensor : public GPUObject, public GpuSpatialTensor {
33 MetalSpatialTensor() in MetalSpatialTensor() function
38 MetalSpatialTensor(id<MTLBuffer> buffer, id<MTLTexture> texture,
41 MetalSpatialTensor(id<MTLBuffer> buffer, id<MTLTexture> texture,
46 MetalSpatialTensor(MetalSpatialTensor&& tensor);
47 MetalSpatialTensor& operator=(MetalSpatialTensor&& tensor);
48 MetalSpatialTensor(const MetalSpatialTensor&) = delete;
49 MetalSpatialTensor& operator=(const MetalSpatialTensor&) = delete;
51 ~MetalSpatialTensor() override { Release(); } in ~MetalSpatialTensor()
108 MetalSpatialTensor* result);
[all …]
Dmetal_spatial_tensor.cc156 MetalSpatialTensor* result) { in CreateTensor()
164 *result = MetalSpatialTensor(buffer, texture, memory_owner, memory_owner, in CreateTensor()
170 MetalSpatialTensor::MetalSpatialTensor(id<MTLBuffer> buffer, in MetalSpatialTensor() function in tflite::gpu::metal::MetalSpatialTensor
183 MetalSpatialTensor::MetalSpatialTensor(id<MTLBuffer> buffer, in MetalSpatialTensor() function in tflite::gpu::metal::MetalSpatialTensor
196 MetalSpatialTensor::MetalSpatialTensor(MetalSpatialTensor&& tensor) in MetalSpatialTensor() function in tflite::gpu::metal::MetalSpatialTensor
206 MetalSpatialTensor& MetalSpatialTensor::operator=(MetalSpatialTensor&& tensor) { in operator =()
219 void MetalSpatialTensor::Release() { in Release()
228 absl::Status MetalSpatialTensor::GetGPUResources( in GetGPUResources()
299 int3 MetalSpatialTensor::GetFullTensorRegion() const { in GetFullTensorRegion()
315 absl::Status MetalSpatialTensor::IsValid(const BHWC& shape) const { in IsValid()
[all …]
Dinference_context.h145 MetalSpatialTensor* GetTensor(ValueId tensor_id);
211 std::map<ValueId, MetalSpatialTensor> preallocated_tensors_;
214 std::map<ValueId, MetalSpatialTensor> const_tensors_;
218 std::vector<MetalSpatialTensor>
222 std::map<ValueId, MetalSpatialTensor> strong_shape_tensors_;
Dcompute_task.h64 void SetSrcTensor(MetalSpatialTensor* tensor, int index);
66 void SetDstTensor(MetalSpatialTensor* tensor, int index);
Dcompute_task.cc193 dynamic_cast<const MetalSpatialTensor*>(operation_->src_[i]); in UpdateParams()
202 dynamic_cast<const MetalSpatialTensor*>(operation_->dst_[i]); in UpdateParams()
230 void ComputeTask::SetSrcTensor(MetalSpatialTensor* tensor, int index) { in SetSrcTensor()
236 void ComputeTask::SetDstTensor(MetalSpatialTensor* tensor, int index) { in SetDstTensor()
Dmetal_spatial_tensor_test.mm56 tflite::gpu::metal::MetalSpatialTensor tensor;
87 tflite::gpu::metal::MetalSpatialTensor tensor;
Dmetal_arguments.cc154 MetalSpatialTensor gpu_tensor; in CreateMetalObject()
156 *result = absl::make_unique<MetalSpatialTensor>(std::move(gpu_tensor)); in CreateMetalObject()
Dinference_context.cc384 MetalSpatialTensor* InferenceContext::GetTensor(ValueId tensor_id) { in GetTensor()
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/kernels/
Dtest_util.cc69 std::vector<MetalSpatialTensor> src(src_cpu.size()); in ExecuteGPUOperation()
81 std::vector<MetalSpatialTensor> dst(dst_cpu.size()); in ExecuteGPUOperation()