Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/cc/gradients/
Dmath_grad_test.cc709 class NaryGradTest : public ::testing::Test { class
711 NaryGradTest() : scope_(Scope::NewRootScope().WithDevice("/cpu:0")) {} in NaryGradTest() function in tensorflow::__anonf4b1ef9d0111::NaryGradTest
734 TEST_F(NaryGradTest, Sum) { in TEST_F() argument
743 TEST_F(NaryGradTest, Mean) { in TEST_F() argument
752 TEST_F(NaryGradTest, Min) { in TEST_F() argument
763 TEST_F(NaryGradTest, Max) { in TEST_F() argument
774 TEST_F(NaryGradTest, MinMulti) { in TEST_F() argument
792 TEST_F(NaryGradTest, MaxMulti) { in TEST_F() argument
801 TEST_F(NaryGradTest, AddN) { in TEST_F() argument
811 TEST_F(NaryGradTest, Add) { in TEST_F() argument
[all …]