Searched refs:first_tensor_shape (Results 1 – 3 of 3) sorted by relevance
95 auto const &first_tensor_shape = first_tensor.sizes(); in can_use_native_serial_stack_impl() local100 "stack expects each tensor to be equal size, but got ", first_tensor_shape, in can_use_native_serial_stack_impl()
1440 first_tensor_shape = g.op("Shape", first_tensor)1441 first_tensor_dim = g.op("Size", first_tensor_shape)
816 const auto first_tensor_shape = inputs[0].sizes(); in varStackFastOut() local820 shape == first_tensor_shape, in varStackFastOut()822 first_tensor_shape, in varStackFastOut()