Searched refs:mutate_index (Results 1 – 4 of 4) sorted by relevance
| /dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/executor/ |
| H A D | attach_op_execs_pass.cc | 290 std::vector<uint32_t> mutate_index; in CreateOpExecs() local
292 mutate_index = fmutate_inputs[op](inode.source->attrs); in CreateOpExecs()
328 exec_type, mutate_index); in CreateOpExecs()
350 mutate_index); in CreateOpExecs()
360 inode.source->attrs, dispatch_modes[i], fcompute, exec_type, mutate_index); in CreateOpExecs()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/src/executor/ |
| H A D | attach_op_execs_pass.cc | 290 std::vector<uint32_t> mutate_index; in CreateOpExecs() local
292 mutate_index = fmutate_inputs[op](inode.source->attrs); in CreateOpExecs()
328 exec_type, mutate_index); in CreateOpExecs()
350 mutate_index); in CreateOpExecs()
360 inode.source->attrs, dispatch_modes[i], fcompute, exec_type, mutate_index); in CreateOpExecs()
|
| /dports/lang/halide/Halide-release_2019_08_27-2654-g664dc4993/src/ |
| H A D | FuseGPUThreadLoops.cpp | 510 Expr mutate_index(SharedAllocation *alloc, const Expr &index) { in mutate_index() function in Halide::Internal::ExtractSharedAndHeapAllocations
525 Expr index = mutate_index(alloc, op->index); in visit()
539 Expr index = mutate_index(alloc, op->index); in visit()
|
| H A D | VectorizeLoops.cpp | 277 Expr mutate_index(const string &a, Expr index) { in mutate_index() function in Halide::Internal::__anon4b0b2d310111::RewriteAccessToVectorAlloc
295 return Load::make(op->type, op->name, mutate_index(op->name, op->index), in visit()
300 return Store::make(op->name, mutate(op->value), mutate_index(op->name, op->index), in visit()
|