Searched refs:getPhysicalToLogicalMap (Results 1 – 9 of 9) sorted by relevance
81 return self_physical.getPhysicalToLogicalMap().apply(result); in sum_batching_rule()101 return physical_args[0].getPhysicalToLogicalMap().apply(result); in binary_pointwise_batching_rule()106 return other_physical.getPhysicalToLogicalMap().apply(result); in binary_pointwise_batching_rule()111 return self_physical.getPhysicalToLogicalMap().apply(result); in binary_pointwise_batching_rule()152 return physical_args[0].getPhysicalToLogicalMap().apply(result); in binary_pointwise_batching_rule()167 return self_physical.getPhysicalToLogicalMap().apply(result); in expand_batching_rule()190 return self_physical.getPhysicalToLogicalMap().apply(result); in expand_batching_rule()197 self_physical.getPhysicalToLogicalMap().applyInplace(result); in chunk_batching_rule()204 return self_physical.getPhysicalToLogicalMap().apply(result); in clamp_batching_rule()210 return self_physical.getPhysicalToLogicalMap().apply(result); in clamp_min_batching_rule()[all …]
141 VmapPhysicalToLogicalMap getPhysicalToLogicalMap() const;
282 VmapPhysicalToLogicalMap VmapPhysicalView::getPhysicalToLogicalMap() const { in getPhysicalToLogicalMap() function in at::VmapPhysicalView
374 … input_physical_views.front().getPhysicalToLogicalMap().apply(flat_output.view(output_sizes))); in batchedTensorForLoopFallback()
245 self_physical.getPhysicalToLogicalMap().applyInplace(result); in split_batching_rule()257 self_physical.getPhysicalToLogicalMap().applyInplace(result); in split_with_sizes_batching_rule()269 self_physical.getPhysicalToLogicalMap().applyInplace(result); in split_with_sizes_copy_batching_rule()281 self_physical.getPhysicalToLogicalMap().applyInplace(result); in unbind_batching_rule()403 return physical_view.getPhysicalToLogicalMap().apply(result); in as_strided_batching_rule()606 return physical_views[0].getPhysicalToLogicalMap().apply(result); in block_diag_batching_rule()624 return physical_views[0].getPhysicalToLogicalMap().apply(result); in stack_batching_rule()692 return physical_view.getPhysicalToLogicalMap().apply(result); in new_empty_strided_batching_rule()
144 VmapPhysicalToLogicalMap getPhysicalToLogicalMap() const;
214 VmapPhysicalToLogicalMap VmapPhysicalView::getPhysicalToLogicalMap() const { in getPhysicalToLogicalMap() function in at::functorch::VmapPhysicalView
395 … input_physical_views.front().getPhysicalToLogicalMap().apply(flat_output.view(output_sizes))); in batchedTensorForLoopFallback()
319 auto result = physical_view.getPhysicalToLogicalMap().apply(physical); in TEST()330 auto result = physical_view.getPhysicalToLogicalMap().apply(physical); in TEST()341 auto result = physical_view.getPhysicalToLogicalMap().apply(physical); in TEST()