Searched refs:set_batchnorm_attributes (Results 1 – 1 of 1) sorted by relevance
848 auto set_batchnorm_attributes = [&](auto op) -> StatusOr<Operation*> { in EmitDnnBatchNorm() local871 return set_batchnorm_attributes(fwd_training); in EmitDnnBatchNorm()878 return set_batchnorm_attributes(backward); in EmitDnnBatchNorm()885 return set_batchnorm_attributes(fwd_inference); in EmitDnnBatchNorm()