Searched refs:set_cudnn_batchnorm_forward_training_metadata (Results 1 – 3 of 3) sorted by relevance
127 void set_cudnn_batchnorm_forward_training_metadata(const string& c) { in set_cudnn_batchnorm_forward_training_metadata() function
6690 default_options_.set_cudnn_batchnorm_forward_training_metadata( in TEST_F()6725 default_options_.set_cudnn_batchnorm_forward_training_metadata( in TEST_F()6758 default_options_.set_cudnn_batchnorm_forward_training_metadata( in TEST_F()6795 default_options_.set_cudnn_batchnorm_forward_training_metadata( in TEST_F()6824 default_options_.set_cudnn_batchnorm_forward_training_metadata( in TEST_F()6852 default_options_.set_cudnn_batchnorm_forward_training_metadata( in TEST_F()
146 options.set_cudnn_batchnorm_forward_training_metadata( in OptimizeHloConvolutionCanonicalization()