Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding_test.cc156 Tensor mul_const_input(DT_FLOAT, mul_const_input_shape); in MulConvPushDownTest() local
157 for (int i = 0; i < mul_const_input.NumElements(); ++i) { in MulConvPushDownTest()
158 mul_const_input.flat<float>()(i) = static_cast<float>(i + 3); in MulConvPushDownTest()
161 ops::Const(s.WithOpName("c"), Input::Initializer(mul_const_input)); in MulConvPushDownTest()