Searched defs:ConcatKernel (Results 1 – 2 of 2) sorted by relevance
775 struct ConcatKernel final : OperatorKernel { struct776 explicit ConcatKernel(std::string prefix): prefix_(std::move(prefix)) {} in ConcatKernel() function778 std::string operator()(const Tensor& tensor1, std::string a, const std::string& b, int64_t c) { in operator ()()782 std::string prefix_;
3786 class ConcatKernel(NopKernel): class