Home
last modified time | relevance | path

Searched defs:first_tensor_shape (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/aten/src/ATen/native/cpu/
DSerialStackImpl.h95 auto const &first_tensor_shape = first_tensor.sizes(); in can_use_native_serial_stack_impl() local
/external/pytorch/torch/csrc/jit/runtime/static/
Dops.cpp816 const auto first_tensor_shape = inputs[0].sizes(); in varStackFastOut() local