Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/cc/gradients/
Ddata_flow_grad_test.cc60 TensorShape d1_shape({2}); in TEST_F() local
64 Placeholder(scope_, DT_FLOAT, Placeholder::Shape(d1_shape)), in TEST_F()
68 RunTest(data, {d1_shape, d2_shape}, {y}, {y_shape}); in TEST_F()