Searched defs:kIdxWorkspace (Results 1 – 1 of 1) sorted by relevance
302 const int kIdxInput = 0, kIdxOutput = 0, kIdxWorkspace = 1; member in tensorflow::MklLRNOp649 kIdxWorkspace = 3, kIdxOutput = 0; member in tensorflow::MklLRNGradOp