Searched defs:numTensors (Results 1 – 5 of 5) sorted by relevance
67 AsyncInputIsOutputTest(const std::string& path, int numTensors) in AsyncInputIsOutputTest()129 AsyncAllreduceTest(const std::string& path, int numTensors) in AsyncAllreduceTest()155 AsyncBroadcastTest(const std::string& path, int numTensors) in AsyncBroadcastTest()185 size_t numTensors = 2) { in runAsyncAllreduceTest()225 size_t numTensors = 1) { in runAsyncBroadcastTest()
61 size_t numTensors = 5; variable
251 size_t numTensors = tpDescriptor.tensors.size(); in tensorpipeAllocate() local
144 size_t numTensors() const { in numTensors() function
6956 void test_stack(const at::IntArrayRef input_shape, int64_t dim, int numTensors) { in test_stack()