Searched defs:store_b (Results 1 – 6 of 6) sorted by relevance
| /external/pytorch/test/cpp/tensorexpr/ |
| D | test_aten.cpp | 26 StmtPtr store_b = b_buf.store({index}, to_float); in TEST() local
53 StmtPtr store_b = b_buf.store({index}, to_float); in TEST() local
80 StmtPtr store_b = b_buf.store({index}, to_float); in TEST() local
616 StmtPtr store_b = b_buf.store({index}, FloatImm::make(1.0f) / load_a); in testATenreciprocal() local
642 StmtPtr store_b = b_buf.store({index}, Max::make(load_a, 0, false)); in TEST() local
668 StmtPtr store_b = b_buf.store( in TEST() local
696 StmtPtr store_b = b_buf.store({index}, log(load_a)); in TEST() local
722 StmtPtr store_b = b_buf.store({index}, fast_log(load_a)); in TEST() local
753 StmtPtr store_b = b_buf.store({index}, fast_tanh(load_a)); in TEST() local
784 StmtPtr store_b = b_buf.store({index}, fast_sigmoid(load_a)); in TEST() local
[all …]
|
| D | tutorial.cpp | 164 StmtPtr store_b = Store::make(B, {i, j}, A.load(i, j)); in main() local
|
| D | test_expr.cpp | 198 StmtPtr store_b = b_buf.store({0}, var); in TEST() local
|
| D | test_cuda.cpp | 540 StorePtr store_b = b_buf.store({0}, add_value); in TEST() local
|
| D | test_llvm.cpp | 239 StmtPtr store_b = b_buf.store({index}, fast_log(load_a)); in TEST() local
|
| D | test_loopnest.cpp | 4234 auto store_b = For::make(j, 0, N, Store::make(b, {j}, Load::make(a, {j}))); in TEST() local
|