Searched refs:foreachTensorInplaceWithFlag (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/aten/src/ATen/functorch/ |
| D | Interpreter.h | 198 void foreachTensorInplaceWithFlag(std::vector<IValue>& args, int64_t begin, int64_t end,
|
| D | DynamicLayer.cpp | 305 foreachTensorInplaceWithFlag(args, begin, end, std::bitset<64>(), func_with_bool); in foreachTensorInplace()
308 void foreachTensorInplaceWithFlag(std::vector<IValue>& args, int64_t begin, int64_t end, in foreachTensorInplaceWithFlag() function
|
| D | ADInterpreters.cpp | 186 …foreachTensorInplaceWithFlag(*stack, static_cast<int64_t>(stack->size() - ret_size), static_cast<i… in autogradBasedTransformSendToNext()
|