Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper_test.cc351 auto bias_t = GenerateRandomTensor<DT_FLOAT>({128}); in TEST_F() local
355 item.feed = {{"input", input_t}, {"filter", filter_t}, {"bias", bias_t}}; in TEST_F()
415 auto bias_t = GenerateTensorWithSetRandom<DTYPE>({64}); in RunTest() local
419 item.feed = {{"lhs", lhs_t}, {"rhs", rhs_t}, {"bias", bias_t}}; in RunTest()
499 auto bias_t = GenerateRandomTensor<DT_FLOAT>({128}); in TEST_F() local
503 item.feed = {{"input", input_t}, {"filter", filter_t}, {"bias", bias_t}}; in TEST_F()
586 auto bias_t = GenerateRandomTensor<DT_FLOAT>({128}); in TEST_F() local
590 item.feed = {{"input", input_t}, {"filter", filter_t}, {"bias", bias_t}}; in TEST_F()
688 auto bias_t = GenerateTensorWithSetRandom<DTYPE>({64}); in RunTest() local
692 item.feed = {{"lhs", lhs_t}, {"rhs", rhs_t}, {"bias", bias_t}}; in RunTest()
[all …]
Dmkl_remapper_test.cc257 auto bias_t = GenerateRandomTensor<DT_FLOAT>({3}); \
262 {"input", input_t}, {"filter", filter_t}, {"bias", bias_t}}; \
Dconstant_folding_test.cc969 auto bias_t = GenerateRandomTensor<DT_FLOAT>(TensorShape({2})); in TEST_F() local
973 {{"x", x_t}, {"y", y_t}, {"a", a_t}, {"b", b_t}, {"bias", bias_t}}); in TEST_F()
977 {{"x", x_t}, {"y", y_t}, {"a", a_t}, {"b", b_t}, {"bias", bias_t}}); in TEST_F()
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_benchmark_test.cc116 Tensor bias_t = MakeRandomTensor<T>({out_depth}); in Conv2DWithBias() local
117 Node* bias = test::graph::Constant(graph, bias_t, "bias"); in Conv2DWithBias()
231 Tensor bias_t = MakeRandomTensor<T>({out_depth}); local
235 Node* bias = test::graph::Constant(graph, bias_t, "bias");