Skip to content

Latest commit

 

History

History
107 lines (99 loc) · 42.6 KB

README.md

File metadata and controls

107 lines (99 loc) · 42.6 KB

ai-operator-standard

优化器算子接口

框架版本:

  • Tensorflow:2.9.0
  • PyTorch:1.11.0
算子 TensorFlow PyTorch MindSpore Paddle
sgd tf.keras.optimizers.SGD torch.optim.SGD
momentum tf.keras.optimizers.SGD torch.optim.SGD
adagrad tf.keras.optimizers.Adagrad torch.optim.Adagrad
adadelta tf.keras.optimizers.Adadelta torch.optim.Adadelta
rmsprop tf.keras.optimizers.RMSprop torch.optim.RMSprop
centeredrmsprop tf.raw_ops.ApplyCenteredRMSProp /
adam tf.keras.optimizers.Adam torch.optim.Adam
adamax tf.keras.optimizers.Adamax torch.optim.Adamax

图像处理算子接口

框架版本:

  • Tensorflow:2.9.0
  • PyTorch:1.11.0
  • torchvision:0.12
算子 TensorFlow PyTorch MindSpore Paddle
adjust_brightness tf.image.adjust_brightness torchvision.transforms.functional.adjust_brightness
adjust_contrast tf.image.adjust_contrast torchvision.transforms.functional.adjust_contrast
adjust_gamma tf.image.adjust_gamma torchvision.transforms.functional.adjust_gamma
adjust_hue tf.image.adjust_hue torchvision.transforms.functional.adjust_hue
adjust_saturation tf.image.adjust_saturation torchvision.transforms.functional.adjust_saturation
random_brightness tf.image.random_brightness /
random_contrast tf.image.random_contrast /
random_hue tf.image.random_hue /
random_saturation tf.image.random_saturation /
rgb_to_grayscale tf.image.rgb_to_grayscale torchvision.transforms.functional.rgb_to_grayscale
grayscale_to_rgb tf.image.grayscale_to_rgb /
rgb_to_hsv tf.image.rgb_to_hsv /
hsv_to_rgb tf.image.hsv_to_rgb /
rgb_to_yuv tf.image.rgb_to_yuv /
yuv_to_rgb tf.image.yuv_to_rgb /
rgb_to_yiq tf.image.rgb_to_yiq /
yiq_to_rgb tf.image.yiq_to_rgb /
convert_image_dtype tf.image.convert_image_dtype torchvision.transforms.functional.convert_image_dtype
horizontal_flip tf.image.flip_left_right torchvision.transforms.functional.hflip
vertical_flip tf.image.flip_up_down torchvision.transforms.functional.vflip
random_horizontal_flip tf.image.random_flip_left_right torchvision.transforms.RandomHorizontalFlip
random_vertical_flip tf.image.random_flip_up_down torchvision.transforms.RandomVerticalFlip
image_transpose tf.image.transpose torch.transpose
rot90 tf.image.rot90 torch.rot90
center_crop tf.image.central_crop torchvision.transforms.CenterCrop
random_crop tf.image.random_crop torchvision.transforms.RandomCrop
crop_to_bounding_box tf.image.crop_to_bounding_box /
pad_to_bounding_box tf.image.pad_to_bounding_box torchvision.transforms.Pad
resize_with_crop_or_pad tf.image.resize_with_crop_or_pad torchvision.transforms.functional.resized_crop
resize tf.image.resize torchvision.transforms.functional.resize
Gaussian_blur / torchvision.transforms.functional.gaussian_blur
image_gradients tf.image.image_gradients /
image_normalize / torchvision.transforms.functional.normalize

分布式训练算子接口

框架版本:

  • Tensorflow:2.9.0
  • PyTorch:1.11.0
算子 TensorFlow PyTorch MindSpore Paddle
comm_init tf.raw_ops.CollectiveInitializeCommunicator torch.distributed.init_process_group
comm_create_by_ranks / torch.distributed.new_group
comm_dup / /
comm_destroy / /
comm_get_size / torch.distributed.get_world_size
comm_get_rank / torch.distributed.get_rank
get_async_op_status / is_completed
wait / wait
comm_send tf.raw_ops.Send torch.distributed.send
comm_recv tf.raw_ops.Recv torch.distributed.recv
comm_broadcast tf.raw_ops.NcclBroadcast torch.distributed.broadcast
comm_all_reduce tf.distribute.NcclAllReduce torch.distributed.all_reduce
comm_reduce tf.raw_ops.NcclReduce torch.distributed.reduce
comm_all_gather / torch.distributed.all_gather
comm_gather tf.raw_ops.CollectiveGatherV2 torch.distributed.gather
comm_scatter / torch.distributed.scatter
comm_reduce_scatter / torch.distributed.reduce_scatter
comm_all_to_all tf.raw_ops.AllToAll torch.distributed.all_to_all
comm_barrier / torch.distributed.barrier
comm_send_async / torch.distributed.isend
comm_recv_async / torch.distributed.irecv
comm_broadcast_async / torch.distributed.broadcast
comm_all_reduce_async / torch.distributed.all_reduce
comm_reduce_async / torch.distributed.reduce
comm_all_gather_async / torch.distributed.all_gather
comm_gather_async / torch.distributed.gather
comm_scatter_async / torch.distributed.scatter
comm_reduce_scatter_async / torch.distributed.reduce_scatter
comm_all_to_all_async / torch.distributed.all_to_all
comm_barrier_async / torch.distributed.barrier