Home
last modified time | relevance | path

Searched refs:MetalArguments (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dmetal_arguments.h34 class MetalArguments : public ArgumentsBinder {
36 MetalArguments() = default;
42 MetalArguments(MetalArguments&& args) = default;
43 MetalArguments& operator=(MetalArguments&& args) = default;
44 MetalArguments(const MetalArguments&) = delete;
45 MetalArguments& operator=(const MetalArguments&) = delete;
Dmetal_arguments.cc177 constexpr char MetalArguments::kArgsPrefix[];
179 absl::Status MetalArguments::Init( in Init()
232 std::string MetalArguments::ScalarArgumentsToStructWithScalarFields( in ScalarArgumentsToStructWithScalarFields()
297 std::string MetalArguments::ScalarArgumentsToStructWithVec4Fields( in ScalarArgumentsToStructWithVec4Fields()
372 absl::Status MetalArguments::SetInt(const std::string& name, int value) { in SetInt()
386 absl::Status MetalArguments::SetFloat(const std::string& name, float value) { in SetFloat()
401 absl::Status MetalArguments::SetHalf(const std::string& name, half value) { in SetHalf()
416 absl::Status MetalArguments::SetObjectRef(const std::string& name, in SetObjectRef()
428 void MetalArguments::Encode(id<MTLComputeCommandEncoder> encoder, in Encode()
458 absl::Status MetalArguments::AllocateObjects(const Arguments& args, in AllocateObjects()
[all …]
Dcompute_task.h77 MetalArguments metal_args_;
/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Darguments.h37 class MetalArguments; variable
79 friend class metal::MetalArguments;