Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/c/eager/
Dunified_api_test.cc38 bool UseFunction() const { return std::get<2>(GetParam()); } in UseFunction() function in tensorflow::__anond654ac910111::UnifiedAPI
55 if (UseFunction() && UseMlir()) { in TEST_P()
79 /*use_function=*/UseFunction()); in TEST_P()
105 if (UseFunction() && UseMlir()) { in TEST_P()
132 /*use_function=*/UseFunction()); in TEST_P()
137 if (!UseFunction()) { in TEST_P()
162 if (!UseFunction()) { in TEST_P()
Dgradient_checker_test.cc111 bool UseFunction() const { return std::get<2>(GetParam()); } in UseFunction() function in tensorflow::gradients::internal::__anonc457a2360111::GradientCheckerTest
139 UseFunction())); in TEST_P()
162 UseFunction())); in TEST_P()
/external/tensorflow/tensorflow/c/experimental/gradients/
Dmath_grad_test.cc111 bool UseFunction() const { return std::get<2>(GetParam()); } in UseFunction() function in tensorflow::gradients::internal::__anon87b76e350111::CppGradients
138 immediate_execution_ctx_.get(), {x.get(), y.get()}, UseFunction())); in TEST_P()
156 immediate_execution_ctx_.get(), {x.get()}, UseFunction())); in TEST_P()
202 immediate_execution_ctx_.get(), {A.get(), B.get()}, UseFunction())); in TEST_P()
222 immediate_execution_ctx_.get(), {x.get()}, UseFunction())); in TEST_P()
240 immediate_execution_ctx_.get(), {x.get()}, UseFunction())); in TEST_P()
267 immediate_execution_ctx_.get(), {x.get(), y.get()}, UseFunction())); in TEST_P()
294 immediate_execution_ctx_.get(), {x.get(), y.get()}, UseFunction())); in TEST_P()
312 immediate_execution_ctx_.get(), {x.get()}, UseFunction())); in TEST_P()
341 {x.get(), y.get()}, UseFunction())); in TEST_P()
[all …]
Dnn_grad_test.cc153 bool UseFunction() const { return std::get<2>(GetParam()); } in UseFunction() function in tensorflow::gradients::internal::__anonb6b0f6070111::CppGradients
169 ReluModel, ReluGradModel, ctx_.get(), {X.get()}, UseFunction())); in TEST_P()
183 absl::MakeSpan(outputs), UseFunction()); in TEST_P()
191 if (UseFunction()) { in TEST_P()
223 {X.get(), Y.get()}, UseFunction())); in TEST_P()
227 if (UseFunction() && UseMlir()) { in TEST_P()
256 UseFunction())); in TEST_P()