Searched refs:NumMatrixInputs (Results 1 – 3 of 3) sorted by relevance
51 int NumMatrixInputs(const OpKernelContext* context) const final { return 2; } in NumMatrixInputs() function
50 virtual int NumMatrixInputs(const OpKernelContext* context) const { in NumMatrixInputs() function
121 for (int i = 0; i < NumMatrixInputs(context); ++i) { in AnalyzeInputs()