experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Use internal linkage for classes defined in compute_task.mm.

Browse Source 
PiperOrigin-RevId: 321461539
Change-Id: Ie6b9755897088b5363c62d9ab8e8d93ca89af9a8
This commit is contained in:
A. Unique TensorFlower 2020-07-15 16:31:29 -07:00 committed by TensorFlower Gardener
parent c2f5a7f68d
commit 381c6e98d2
1 changed files with 21 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
tensorflow/lite/delegates/gpu/metal/compute_task.mm
View File

@ -39,22 +39,28 @@ using ::tflite::gpu::metal::UniformsFunction;
using ::tflite::gpu::uint3;
using ::tflite::gpu::ValueId;
@implementation TFLComputeTask {
struct InputBuffer {
ValueId uid;
id<MTLBuffer> metalHandle;
};
struct OutputBuffer {
ValueId uid;
id<MTLBuffer> metalHandle;
OutputDimensions dimensionsFunction;
std::vector<ValueId> alias;
};
struct UniformBuffer {
std::vector<uint8_t> data;
UniformsFunction dataFunction;
};
namespace {
struct InputBuffer {
ValueId uid;
id<MTLBuffer> metalHandle;
};
struct OutputBuffer {
ValueId uid;
id<MTLBuffer> metalHandle;
OutputDimensions dimensionsFunction;
std::vector<ValueId> alias;
};
struct UniformBuffer {
std::vector<uint8_t> data;
UniformsFunction dataFunction;
};
} // namespace
@implementation TFLComputeTask {
id<MTLComputePipelineState> _program;
std::vector<InputBuffer> _inputBuffers;
std::vector<OutputBuffer> _outputBuffers;