Searched refs:transpose_folding (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | transpose_folding_test.cc | 47 TransposeFolding transpose_folding( in FoldTranspose() local
56 EXPECT_IS_OK(transpose_folding.Run(module).status()); in FoldTranspose()
96 TransposeFolding transpose_folding( in TEST_F() local
105 TF_ASSERT_OK_AND_ASSIGN(bool changed, transpose_folding.Run(module.get())); in TEST_F()
124 TransposeFolding transpose_folding( in TEST_F() local
133 TF_ASSERT_OK_AND_ASSIGN(bool changed, transpose_folding.Run(module.get())); in TEST_F()
|
| D | BUILD | 3321 name = "transpose_folding",
3322 srcs = ["transpose_folding.cc"],
3323 hdrs = ["transpose_folding.h"],
3341 ":transpose_folding",
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion_test.cc | 184 TransposeFolding transpose_folding( in TEST_F() local
190 TF_ASSERT_OK_AND_ASSIGN(bool changed, transpose_folding.Run(module.get())); in TEST_F()
214 TransposeFolding transpose_folding( in TEST_F() local
220 TF_ASSERT_OK_AND_ASSIGN(bool changed, transpose_folding.Run(module.get())); in TEST_F()
245 TransposeFolding transpose_folding( in TEST_F() local
251 TF_ASSERT_OK_AND_ASSIGN(bool changed, transpose_folding.Run(module.get())); in TEST_F()
|
| D | BUILD | 140 "//tensorflow/compiler/xla/service:transpose_folding",
722 "//tensorflow/compiler/xla/service:transpose_folding",
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | BUILD | 866 "//tensorflow/compiler/xla/service:transpose_folding",
|