Home
last modified time | relevance | path

Searched defs:flat_output (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/io_ops/
Dparse_single_example_op_test.py66 flat_output): argument
/external/pytorch/aten/src/ATen/
DLegacyBatchedFallback.cpp361 auto flat_output = safeStack(shards); in batchedTensorForLoopFallback() local
/external/pytorch/aten/src/ATen/functorch/
DBatchedFallback.cpp382 auto flat_output = safeStack(shards); in batchedTensorForLoopFallback() local
/external/pytorch/torch/csrc/distributed/c10d/
DProcessGroupUCC.cpp899 std::vector<at::Tensor> flat_output(outputTensors.size()); in allgather() local