Home
last modified time | relevance | path

Searched refs:UseMlir (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/c/eager/
Dunified_api_test.cc37 bool UseMlir() const { return strcmp(std::get<0>(GetParam()), "mlir") == 0; } in UseMlir() function in tensorflow::__anond654ac910111::UnifiedAPI
55 if (UseFunction() && UseMlir()) { in TEST_P()
105 if (UseFunction() && UseMlir()) { in TEST_P()
140 if (UseMlir()) { in TEST_P()
165 if (UseMlir()) { in TEST_P()
Dgradient_checker_test.cc110 bool UseMlir() const { return strcmp(std::get<0>(GetParam()), "mlir") == 0; } in UseMlir() function in tensorflow::gradients::internal::__anonc457a2360111::GradientCheckerTest
/external/tensorflow/tensorflow/c/experimental/gradients/
Dnn_grad_test.cc152 bool UseMlir() const { return strcmp(std::get<0>(GetParam()), "mlir") == 0; } in UseMlir() function in tensorflow::gradients::internal::__anonb6b0f6070111::CppGradients
227 if (UseFunction() && UseMlir()) { in TEST_P()
Dmath_grad_test.cc110 bool UseMlir() const { return strcmp(std::get<0>(GetParam()), "mlir") == 0; } in UseMlir() function in tensorflow::gradients::internal::__anon87b76e350111::CppGradients