Searched defs:GetTypeOfInputTensorFromFwdOp (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ | ||
| H A D | mkl_relu_op.cc | 585 virtual int GetTypeOfInputTensorFromFwdOp() const { return DNNL_ARG_SRC; } in GetTypeOfInputTensorFromFwdOp() function in tensorflow::MklReluGradOpBase |