Searched refs:mul_values_data (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fold_batch_norms_test.cc | 59 Tensor mul_values_data(DT_FLOAT, TensorShape({2})); in TestFoldBatchNormsConv2D() local
60 test::FillValues<float>(&mul_values_data, {2.0f, 3.0f}); in TestFoldBatchNormsConv2D()
62 Input::Initializer(mul_values_data)); in TestFoldBatchNormsConv2D()
110 Tensor mul_values_data(DT_FLOAT, TensorShape({4})); in TestFoldBatchNormsDepthwiseConv2dNative() local
111 test::FillValues<float>(&mul_values_data, {2.0f, 3.0f, 4.0f, 5.0f}); in TestFoldBatchNormsDepthwiseConv2dNative()
113 Input::Initializer(mul_values_data)); in TestFoldBatchNormsDepthwiseConv2dNative()
161 Tensor mul_values_data(DT_FLOAT, TensorShape({2})); in TestFoldBatchNormsConv2DShared() local
162 test::FillValues<float>(&mul_values_data, {2.0f, 3.0f}); in TestFoldBatchNormsConv2DShared()
164 Input::Initializer(mul_values_data)); in TestFoldBatchNormsConv2DShared()
171 Input::Initializer(mul_values_data)); in TestFoldBatchNormsConv2DShared()
[all …]
|