Searched defs:FusedKernel (Results 1 – 3 of 3) sorted by relevance
14 struct FusedKernel { struct17 FusedKernel( in FusedKernel() argument50 const std::string& name() const { in name()53 const std::string& code() const { in code()56 const std::vector<TensorDesc>& inputDesc() const { in inputDesc()59 const std::vector<TensorDesc>& outputDesc() const { in outputDesc()62 const std::vector<PartitionDesc>& chunkDesc() const { in chunkDesc()65 const std::vector<PartitionDesc>& concatDesc() const { in concatDesc()68 bool hasRandom() const { in hasRandom()76 const std::string code_;[all …]
39 typedef FusedKernelParams<QuantizedStaticPreprocessed, RowMajor> FusedKernel; typedef88 FusedKernel; typedef133 FusedKernel; typedef
56 typedef FusedKernelParams<Kernel, OutputStream> FusedKernel; typedef