Searched refs:create_coordinate_tensors (Results 1 – 1 of 1) sorted by relevance
2923 def create_coordinate_tensors(shape): function2951 … (create_scalar, create_list, create_coordinate_tensors)):2961 if space_fn == create_coordinate_tensors and spacing[0].device != 'cpu':