tvm::backend::opencl Namespace Reference

Functions
TVM_TIRX_DLL const Op &	texture2d_store_op ()
	Store to texture 2d memory.
TVM_TIRX_DLL const Op &	texture2d_load_op ()
	Load from texture 2d memory.
TVM_TIRX_DLL const Op &	nd_mem_alloc_with_scope_op ()
	Create an Nd memory allocation with storage scope.

Function Documentation

nd_mem_alloc_with_scope_op()

TVM_TIRX_DLL const Op & tvm::backend::opencl::nd_mem_alloc_with_scope_op

(

)

Create an Nd memory allocation with storage scope.

Arguments, in order:

• args[0]: storage_scope, The storage scope.
• args[1]: ndim, The number of dimensions.
• args[2]: shape, The shape.
• args[3...]: args, trailing Expr operands.

texture2d_load_op()

TVM_TIRX_DLL const Op & tvm::backend::opencl::texture2d_load_op

(

)

Load from texture 2d memory.

Arguments, in order:

• args[0]: texture, The texture.
• args[1]: x, The input value.
• args[2]: y, The second input value.
• args[3]: z, The third input value.
• args[4]: channel_size, The number of channels.
• args[5]: element_index, The element index within a texture channel.

texture2d_store_op()

TVM_TIRX_DLL const Op & tvm::backend::opencl::texture2d_store_op

(

)

Store to texture 2d memory.

Arguments, in order:

• args[0]: texture, The texture.
• args[1]: x, The input value.
• args[2]: y, The second input value.
• args[3]: z, The third input value.
• args[4]: channel_size, The number of channels.
• args[5]: value, The value to use.