CANN/tensorflow TF Adapter 1.x API参考

TF Adapter 1.x API

【免费下载链接】tensorflowAscend TensorFlow Adapter项目地址: https://gitcode.com/cann/tensorflow

• TF Adapter接口简介

• session配置

• npu_bridge.estimator.npu.npu_config

  • NPURunConfig构造函数
  • RunConfig参数支持说明
  • ProfilingConfig构造函数
  • DumpConfig构造函数
  • MemoryConfig构造函数
  • ExperimentalConfig构造函数

• npu_bridge.estimator.npu.npu_estimator

  • NPUEstimator构造函数
  • NPUEstimatorSpec构造函数

• npu_bridge.estimator.npu.npu_strategy

  • NPUStrategy构造函数

• npu_bridge.estimator.npu.npu_hook

  • NPUCheckpointSaverHook构造函数
  • NPUOutputTensorHook构造函数
  • TellMeStepOrLossHook构造函数

• npu_bridge.estimator.npu.npu_optimizer

  • NPUDistributedOptimizer构造函数
  • NPUOptimizer构造函数
  • KerasDistributeOptimizer构造函数
  • npu_distributed_optimizer_wrapper
  • npu_allreduce

• npu_bridge.estimator.npu.npu_callbacks

  • NPUBroadcastGlobalVariablesCallback构造函数

• npu_bridge.estimator.npu.npu_loss_scale_optimizer

  • NPULossScaleOptimizer构造函数

• npu_bridge.estimator.npu.npu_loss_scale_manager

  • FixedLossScaleManager构造函数
  • ExponentialUpdateLossScaleManager构造函数

• npu_bridge.estimator.npu_ops

  • dropout
  • LARSV2
  • initialize_system
  • shutdown_system
  • npu_onnx_graph_op

• npu_bridge.estimator.npu.npu_rnn

  • npu_dynamic_rnn

• npu_bridge.estimator.npu.npu_dynamic_rnn

  • DynamicRNN构造函数
  • DynamicGRUV2构造函数

• npu_bridge.estimator.npu.npu_scope

  • without_npu_compile_scope
  • keep_dtype_scope
  • npu_weight_prefetch_scope
  • subgraph_multi_dims_scope
  • disable_autofuse

• npu_bridge.estimator.npu.util

  • set_iteration_per_loop
  • create_iteration_per_loop_var
  • load_iteration_per_loop_var
  • set_graph_exec_config
  • keep_tensors_dtypes
  • set_op_input_tensor_multi_dims

• npu_bridge.estimator.npu.keras_to_npu

  • model_to_npu_estimator

• npu_bridge.estimator.npu.npu_plugin

  • set_device_sat_mode

• npu_bridge.scoped_graph_manager.scoped_graph_manager

  • ScopedGraphManager

• npu_bridge.profiler.profiler

  • Profiler构造函数

• npu_bridge.hccl.hccl_ops

  • 简介
  • allreduce
  • allgather
  • broadcast
  • reduce_scatter
  • reduce
  • alltoallv
  • alltoallvc
  • send
  • receive
  • 样例代码
    • 代码示例
    • 样例执行

• TensorFlow 1.15 API支持列表

【免费下载链接】tensorflowAscend TensorFlow Adapter项目地址: https://gitcode.com/cann/tensorflow