Searched refs:get_nested_tensor_structure (Results 1 – 1 of 1) sorted by relevance
332 inline TensorNode get_nested_tensor_structure(at::Tensor tensor) { in get_nested_tensor_structure() function440 impl::map(std::forward<F>(fn), impl::get_nested_tensor_structure(a)...), in map_nested_tensor()