Searched refs:map_nested_tensor (Results 1 – 2 of 2) sorted by relevance
22 …ions for working with NestedTensors. (This is where you will find `map_nested_tensor` which is di…56 …ur main concern and you would like to enable coverage the function `map_nested_tensor` can be foun…
438 inline at::Tensor map_nested_tensor(F&& fn, A... a) { in map_nested_tensor() function