Searched refs:tensor_ops_enabled_ (Results 1 – 1 of 1) sorted by relevance
4226 tensor_ops_enabled_(tensor_ops_enabled), in CudnnLegacyConvRunner()4238 return {algo_id_, tensor_ops_enabled_, workspace_size_}; in MakeAlgorithmDesc()4244 bool tensor_ops_enabled_; member in stream_executor::gpu::CudnnLegacyConvRunner4955 << ", tensor_ops_enabled=" << tensor_ops_enabled_ in operator ()()4967 if (IsTensorMathOpSet(conv_) != tensor_ops_enabled_) { in operator ()()5004 << ", tensor_ops_enabled=" << tensor_ops_enabled_ in operator ()()5028 tensor_ops_enabled_(tensor_ops_enabled), in CudnnLegacyFusedConvRunner()5042 return {algo_id_, tensor_ops_enabled_, workspace_size_}; in MakeAlgorithmDesc()5048 bool tensor_ops_enabled_; member in stream_executor::gpu::CudnnLegacyFusedConvRunner