| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_conv_ops_test.cc | 124 Tensor filter_t = GetRandomFilterTensor(dims); in DefaultConv2D() local
127 Node* filter = test::graph::Constant(graph, filter_t, "filter"); in DefaultConv2D()
146 Tensor filter_t = GetRandomFilterTensor(dims); in MklConv2D() local
149 Node* filter = test::graph::Constant(graph, filter_t, "filter"); in MklConv2D()
173 Tensor filter_t = GetRandomFilterTensor(dims); in DefaultConv2DBwdInput() local
178 Node* filter = test::graph::Constant(graph, filter_t, "filter"); in DefaultConv2DBwdInput()
201 Tensor filter_t = GetRandomFilterTensor(dims); in MklConv2DBwdInput() local
206 Node* filter = test::graph::Constant(graph, filter_t, "filter"); in MklConv2DBwdInput()
236 Tensor filter_t = GetRandomFilterTensor(dims); in DefaultConv2DBwdFilter() local
265 Tensor filter_t = GetRandomFilterTensor(dims); in MklConv2DBwdFilter() local
|
| D | xsmm_conv2d_test.cc | 140 LIBXSMM_VLA_DECL(4, const float, filter_t, filter, nIfm, kh, kw); in naive_conv_fp()
157 LIBXSMM_VLA_ACCESS(4, filter_t, ofm, ifm, kj, ki, nIfm, kh, in naive_conv_fp()
|
| D | conv_ops_test.cc | 1109 Tensor filter_t = MakeRandomTensor({filter_w, filter_h, in_depth, out_depth}); in Conv2D() local
1112 Node* filter = test::graph::Constant(graph, filter_t, "filter"); in Conv2D()
1237 Tensor filter_t = MakeRandomTensor({filter_w, filter_h, in_depth, out_depth}); local
1241 Node* filter = test::graph::Constant(graph, filter_t, "filter");
1269 Tensor filter_t = MakeRandomTensor({filter_w, filter_h, in_depth, out_depth}); local
1276 Node* filter = test::graph::Constant(graph, filter_t, "filter");
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | remapper_test.cc | 125 auto filter_t = GenerateRandomTensor<DT_FLOAT>({1, 1, 3, 128}); in TEST_F() local
130 item.feed = {{"input", input_t}, {"filter", filter_t}, {"bias", bias_t}}; in TEST_F()
189 auto filter_t = GenerateRandomTensor<DT_FLOAT>({1, 1, 3, 128}); in TEST_F() local
194 item.feed = {{"input", input_t}, {"filter", filter_t}, {"bias", bias_t}}; in TEST_F()
259 auto filter_t = GenerateRandomTensor<DT_FLOAT>({1, 1, 3, 128}); in TEST_F() local
267 item.feed = {{"input", input_t}, {"filter", filter_t}, in TEST_F()
337 auto filter_t = GenerateRandomTensor<DT_FLOAT>({1, 1, 3, 128}); in TEST_F() local
345 item.feed = {{"input", input_t}, {"filter", filter_t}, in TEST_F()
414 auto filter_t = GenerateRandomTensor<DT_FLOAT>({1, 1, 3, 128}); in TEST_F() local
419 item.feed = {{"input", input_t}, {"filter", filter_t}, {"bias", bias_t}}; in TEST_F()
|