Home
last modified time | relevance | path

Searched defs:DataFormatVecPermute (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Ddata_format_ops_gpu.cu.cc28 template struct functor::DataFormatVecPermute<GPUDevice, int32>; member in tensorflow::functor
29 template struct functor::DataFormatVecPermute<GPUDevice, int64>; member in tensorflow::functor
H A Ddata_format_ops.h102 struct DataFormatVecPermute { struct
103 void operator()(const Device& d, typename TTypes<T>::ConstFlat x, in operator()
/aosp_15_r20/external/tensorflow/tensorflow/go/op/
H A Dwrappers.go9829 func DataFormatVecPermute(scope *Scope, x tf.Output, optional ...DataFormatVecPermuteAttr) (y tf.Ou… func