Searched defs:MockKernel (Results 1 – 2 of 2) sorted by relevance
15 MockKernel() {} in MockKernel() function
41 struct MockKernel final : OperatorKernel { struct42 MockKernel(bool* called): called_(called) {} in MockKernel() function44 void operator()(Tensor) { in operator ()()48 bool* called_;