Home
last modified time | relevance | path

Searched refs:use_mkl_dnn (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter.cc740 bool use_mkl_dnn = hlo_module_config_.debug_options().xla_cpu_use_mkl_dnn(); in EmitCallToRuntime() local
755 ? (use_mkl_dnn ? runtime::kMKLMatMulF32SymbolName in EmitCallToRuntime()
757 : (use_mkl_dnn in EmitCallToRuntime()
764 ? (use_mkl_dnn ? runtime::kMKLMatMulF64SymbolName in EmitCallToRuntime()
766 : (use_mkl_dnn in EmitCallToRuntime()
Dir_emitter.cc960 bool use_mkl_dnn = in HandleConvolution() local
971 ? (use_mkl_dnn ? runtime::kMKLConvF32SymbolName in HandleConvolution()
974 if (!multi_threaded && use_mkl_dnn) { in HandleConvolution()