Index

_

__init__() (ipu_tensorflow_addons.keras.layers.ConvertFromF8 method) (ipu_tensorflow_addons.keras.optimizers.AdamIpuOptimizer method) (ipu_tensorflow_addons.keras.optimizers.LAMBIpuOptimizer method) (ipu_tensorflow_addons.keras.optimizers.SGDIpuOptimizer method) (ipu_tensorflow_addons.v1.layers.PopnnAUGRU method) (ipu_tensorflow_addons.v1.layers.PopnnDynamicGRU method) (ipu_tensorflow_addons.v1.layers.PopnnGRU method) (ipu_tensorflow_addons.v1.layers.PopnnLSTM method) (tensorflow.python.ipu.data.ops.dataset_ops.BufferDataset method) (tensorflow.python.ipu.distributed.popdist_strategy.PopDistStrategy method) (tensorflow.python.ipu.embedding_ops.HostEmbedding method) (tensorflow.python.ipu.embedding_ops.HostEmbeddingOptimizerSpec method) (tensorflow.python.ipu.embedding_ops.HostEmbeddingSGDGAOptimizerSpec method) (tensorflow.python.ipu.ipu_infeed_queue.IPUInfeedQueue method) (tensorflow.python.ipu.ipu_infeed_queue.IPUIterator method) (tensorflow.python.ipu.ipu_infeed_queue.IPUOwnedIterator method) (tensorflow.python.ipu.ipu_outfeed_queue.IPUOutfeedQueue method) (tensorflow.python.ipu.ipu_outfeed_queue.IPUOutfeedQueueIterator method) (tensorflow.python.ipu.ipu_outfeed_queue.ScopedIPUOutfeedQueue method) (tensorflow.python.ipu.ipu_run_config.RunConfig method) (tensorflow.python.ipu.ipu_session_run_hooks.IPULoggingTensorHook method) (tensorflow.python.ipu.ipu_strategy.IPUExtendedV1 method) (tensorflow.python.ipu.ipu_strategy.IPUStrategyV1 method) (tensorflow.python.ipu.optimizers.CrossReplicaGradientAccumulationOptimizer method) (tensorflow.python.ipu.optimizers.CrossReplicaGradientAccumulationOptimizerV2 method) (tensorflow.python.ipu.optimizers.CrossReplicaOptimizer method) (tensorflow.python.ipu.optimizers.GradientAccumulationOptimizer method) (tensorflow.python.ipu.optimizers.GradientAccumulationOptimizerV2 method) (tensorflow.python.ipu.optimizers.IpuOptimizer method) (tensorflow.python.ipu.optimizers.MapGradientOptimizer method) (tensorflow.python.ipu.optimizers.ShardedOptimizer method) (tensorflow.python.ipu.pipelining_ops.OptimizerFunctionOutput method) (tensorflow.python.ipu.pipelining_ops.PipelineStageOptions method)

__new__() (tensorflow.python.ipu.ipu_run_config.IPURunConfig static method)

A

AdamIpuOptimizer (class in ipu_tensorflow_addons.keras.optimizers) after_run() (tensorflow.python.ipu.ipu_session_run_hooks.IPULoggingTensorHook method) algorithm (tensorflow.python.ipu.config.IPUConfig.scheduling attribute) all_gather() (in module tensorflow.python.ipu.all_to_all_op) (in module tensorflow.python.ipu.within_replica_ops) all_reduce() (in module tensorflow.python.ipu.within_replica_ops) all_to_all() (in module tensorflow.python.ipu.all_to_all_op) allow_recompute (tensorflow.python.ipu.config.IPUConfig attribute) ALSGradientAccumulationOptimizer (class in keras.ipu.optimizers) ALSOptimizer (class in keras.ipu.optimizers) ALSOptimizerAdam (class in keras.ipu.optimizers) ALSOptimizerRMSProp (class in keras.ipu.optimizers)

ALSOptimizerSGD (class in keras.ipu.optimizers) always_rearrange_copies_on_the_host (tensorflow.python.ipu.config.IPUConfig.experimental attribute) apply_gradients() (keras.ipu.optimizers.ALSGradientAccumulationOptimizer method) (keras.ipu.optimizers.ALSOptimizer method) (tensorflow.python.ipu.optimizers.CrossReplicaOptimizer method) (tensorflow.python.ipu.optimizers.GradientAccumulationOptimizer method) (tensorflow.python.ipu.optimizers.GradientAccumulationOptimizerV2 method) (tensorflow.python.ipu.optimizers.IpuOptimizer method) (tensorflow.python.ipu.optimizers.ShardedOptimizer method) assume_equal_across_replicas() (in module tensorflow.python.ipu.cross_replica_ops) AssumeEqualAcrossReplicas (class in ipu_tensorflow_addons.keras.layers) auto_select_ipus (tensorflow.python.ipu.config.IPUConfig attribute) available_memory_proportion (tensorflow.python.ipu.config.IPUConfig.io_tiles attribute)

B

barrier() (in module tensorflow.python.ipu.control_flow_ops) begin() (tensorflow.python.ipu.ipu_session_run_hooks.IPULoggingTensorHook method) BufferDataset (class in tensorflow.python.ipu.data.ops.dataset_ops) build() (ipu_tensorflow_addons.keras.layers.Dense method) (ipu_tensorflow_addons.keras.layers.Dropout method) (ipu_tensorflow_addons.keras.layers.EffectiveTransformer method) (ipu_tensorflow_addons.keras.layers.Embedding method) (ipu_tensorflow_addons.keras.layers.GroupNormalization method) (ipu_tensorflow_addons.keras.layers.InstanceNormalization method) (ipu_tensorflow_addons.keras.layers.LayerNormalization method) (ipu_tensorflow_addons.keras.layers.PopnnGRU method) (ipu_tensorflow_addons.keras.layers.PopnnLSTM method) (ipu_tensorflow_addons.keras.layers.SerialDense method) (ipu_tensorflow_addons.v1.layers.PopnnGRU method) (ipu_tensorflow_addons.v1.layers.PopnnLSTM method)

C

call() (ipu_tensorflow_addons.keras.layers.AssumeEqualAcrossReplicas method) (ipu_tensorflow_addons.keras.layers.ConvertFromF8 method) (ipu_tensorflow_addons.keras.layers.ConvertToF8 method) (ipu_tensorflow_addons.keras.layers.CTCInferenceLayer method) (ipu_tensorflow_addons.keras.layers.CTCLoss method) (ipu_tensorflow_addons.keras.layers.CTCPredictionsLayer method) (ipu_tensorflow_addons.keras.layers.Dense method) (ipu_tensorflow_addons.keras.layers.Dropout method) (ipu_tensorflow_addons.keras.layers.EffectiveTransformer method) (ipu_tensorflow_addons.keras.layers.Embedding method) (ipu_tensorflow_addons.keras.layers.GroupNormalization method) (ipu_tensorflow_addons.keras.layers.InstanceNormalization method) (ipu_tensorflow_addons.keras.layers.LayerNormalization method) (ipu_tensorflow_addons.keras.layers.PopnnGRU method) (ipu_tensorflow_addons.keras.layers.PopnnLSTM method) (ipu_tensorflow_addons.keras.layers.RecomputationCheckpoint method) (ipu_tensorflow_addons.keras.layers.SerialDense method) (ipu_tensorflow_addons.v1.layers.PopnnAUGRU method) (ipu_tensorflow_addons.v1.layers.PopnnDynamicGRU method) (ipu_tensorflow_addons.v1.layers.PopnnDynamicLSTM method) (ipu_tensorflow_addons.v1.layers.PopnnGRU method) (ipu_tensorflow_addons.v1.layers.PopnnLSTM method) cholesky_block_size (tensorflow.python.ipu.config.IPUConfig.optimizations attribute) clear_pass_type (tensorflow.python.ipu.config.IPUConfig.matmuls attribute) codelet_expression_op() (in module tensorflow.python.ipu.custom_ops) combine_embedding_lookups (tensorflow.python.ipu.config.IPUConfig.optimizations attribute) combine_matmuls (tensorflow.python.ipu.config.IPUConfig.optimizations attribute) compilation_poplar_options (tensorflow.python.ipu.config.IPUConfig attribute) compile() (in module tensorflow.python.ipu.ipu_compiler) compile_ipu_code (tensorflow.python.ipu.config.IPUConfig.ipu_model attribute) computational_stages (tensorflow.python.ipu.ipu_pipeline_estimator.IPUPipelineEstimatorSpec attribute)

compute_gradients() (tensorflow.python.ipu.optimizers.IpuOptimizer method) (tensorflow.python.ipu.optimizers.MapGradientOptimizer method) (tensorflow.python.ipu.optimizers.ShardedOptimizer method) compute_output_shape() (ipu_tensorflow_addons.keras.layers.Dropout method) (ipu_tensorflow_addons.keras.layers.Embedding method) (ipu_tensorflow_addons.keras.layers.SerialDense method) configure_ipu_system() (in module tensorflow.python.ipu.config) (tensorflow.python.ipu.config.IPUConfig method) ConvertFromF8 (class in ipu_tensorflow_addons.keras.layers) ConvertToF8 (class in ipu_tensorflow_addons.keras.layers) convolutions (tensorflow.python.ipu.config.IPUConfig attribute) count() (tensorflow.python.ipu.ipu_estimator.IPUEstimatorSpec method) (tensorflow.python.ipu.ipu_pipeline_estimator.IPUPipelineEstimatorSpec method) cpu_user_operation() (in module tensorflow.python.ipu.custom_ops) create_deregister_instruction() (tensorflow.python.ipu.embedding_ops.HostEmbeddingOptimizerSpec method) create_host_embedding() (in module tensorflow.python.ipu.embedding_ops) create_lookup_instruction() (tensorflow.python.ipu.embedding_ops.HostEmbeddingOptimizerSpec method) create_register_instruction() (tensorflow.python.ipu.embedding_ops.HostEmbeddingOptimizerSpec method) create_slot_variables() (tensorflow.python.ipu.embedding_ops.HostEmbeddingOptimizerSpec method) cross_replica_mean() (in module tensorflow.python.ipu.cross_replica_ops) cross_replica_sum() (in module tensorflow.python.ipu.cross_replica_ops) CrossReplicaGradientAccumulationOptimizer (class in tensorflow.python.ipu.optimizers) CrossReplicaGradientAccumulationOptimizerV2 (class in tensorflow.python.ipu.optimizers) CrossReplicaOptimizer (class in tensorflow.python.ipu.optimizers) ctc_beam_search_decoder() (in module tensorflow.python.ipu.nn_ops) ctc_beam_search_decoder_with_log_probs() (in module tensorflow.python.ipu.nn_ops) ctc_loss_v2() (in module tensorflow.python.ipu.nn_ops) ctc_loss_with_log_probs() (in module tensorflow.python.ipu.nn_ops) CTCInferenceLayer (class in ipu_tensorflow_addons.keras.layers) CTCLoss (class in ipu_tensorflow_addons.keras.layers) CTCPredictionsLayer (class in ipu_tensorflow_addons.keras.layers)

D

dataset_benchmark() (in module tensorflow.python.ipu.dataset_benchmark) deleter (tensorflow.python.ipu.ipu_infeed_queue.IPUInfeedQueue property) (tensorflow.python.ipu.ipu_outfeed_queue.IPUOutfeedQueue property) Dense (class in ipu_tensorflow_addons.keras.layers) dependencies() (in module tensorflow.python.ipu.sharding) dequeue() (tensorflow.python.ipu.ipu_outfeed_queue.IPUOutfeedQueue method) dequeued (tensorflow.python.ipu.ipu_infeed_queue.IPUInfeedQueue property)

device_connection (tensorflow.python.ipu.config.IPUConfig attribute) device_mapping (tensorflow.python.ipu.ipu_pipeline_estimator.IPUPipelineEstimatorSpec attribute) DeviceConnectionType (class in tensorflow.python.ipu.config) distributed_batch_norm_replica_group_size (tensorflow.python.ipu.config.norms.norms.experimental attribute) div0 (tensorflow.python.ipu.config.IPUConfig.floating_point_behaviour attribute) dot_strength (tensorflow.python.ipu.config.IPUConfig.optimizations.math attribute) Dropout (class in ipu_tensorflow_addons.keras.layers) dropout() (in module tensorflow.python.ipu.rand_ops)

E

EffectiveTransformer (class in ipu_tensorflow_addons.keras.layers) element_spec (tensorflow.python.ipu.ipu_infeed_queue.IPUIterator property) embedded_runtime_call() (in module tensorflow.python.ipu.embedded_runtime) embedded_runtime_start() (in module tensorflow.python.ipu.embedded_runtime) embedded_runtime_start_and_call() (in module tensorflow.python.ipu.embedded_runtime) embedded_runtime_stop() (in module tensorflow.python.ipu.embedded_runtime) Embedding (class in ipu_tensorflow_addons.keras.layers) embedding_lookup() (in module tensorflow.python.ipu.embedding_ops) enable_dynamic_slice_replacement (tensorflow.python.ipu.config.IPUConfig.optimizations attribute) enable_fast_math (tensorflow.python.ipu.config.IPUConfig.optimizations attribute) enable_gather_simplifier (tensorflow.python.ipu.config.IPUConfig.optimizations attribute) enable_graph_outlining (tensorflow.python.ipu.config.IPUConfig.optimizations attribute) enable_prng_stability (tensorflow.python.ipu.config.IPUConfig.experimental attribute) enable_remote_buffer_embedding (tensorflow.python.ipu.config.IPUConfig.experimental attribute) enable_remote_buffers (tensorflow.python.ipu.config.IPUConfig.device_connection attribute) enable_sharded_gradient_tape() (in module tensorflow.python.ipu.sharding) end() (tensorflow.python.ipu.ipu_session_run_hooks.IPULoggingTensorHook method) enqueue() (tensorflow.python.ipu.ipu_outfeed_queue.IPUOutfeedQueue method) Enum (class in tensorflow.python.ipu.gradient_accumulation) esr (tensorflow.python.ipu.config.IPUConfig.floating_point_behaviour attribute) eval_dir() (tensorflow.python.ipu.ipu_estimator.IPUEstimator method) (tensorflow.python.ipu.ipu_pipeline_estimator.IPUPipelineEstimator method)

eval_metric_ops (tensorflow.python.ipu.ipu_estimator.IPUEstimatorSpec attribute) eval_metrics (tensorflow.python.ipu.ipu_estimator.IPUEstimatorSpec attribute) eval_metrics_fn (tensorflow.python.ipu.ipu_pipeline_estimator.IPUPipelineEstimatorSpec attribute) evaluate() (tensorflow.python.ipu.ipu_estimator.IPUEstimator method) (tensorflow.python.ipu.ipu_pipeline_estimator.IPUPipelineEstimator method) evaluation_hooks (tensorflow.python.ipu.ipu_estimator.IPUEstimatorSpec attribute) (tensorflow.python.ipu.ipu_pipeline_estimator.IPUPipelineEstimatorSpec attribute) executing_eagerly() (in module tensorflow.python.ipu.embedded_runtime) ExecutionProfileType (class in tensorflow.python.ipu.config) experimental (tensorflow.python.ipu.config.IPUConfig attribute) (tensorflow.python.ipu.config.IPUConfig.norms attribute) experimental_application_compile_op() (in module tensorflow.python.ipu.application_compile_op) experimental_export_all_saved_models() (tensorflow.python.ipu.ipu_estimator.IPUEstimator method) (tensorflow.python.ipu.ipu_pipeline_estimator.IPUPipelineEstimator method) export_dataset_to_file() (in module tensorflow.python.ipu.utils) export_inputs_to_file() (in module tensorflow.python.ipu.utils) export_keras() (in module tensorflow.python.ipu.serving) export_pipeline() (in module tensorflow.python.ipu.serving) export_saved_model() (tensorflow.python.ipu.ipu_estimator.IPUEstimator method) (tensorflow.python.ipu.ipu_pipeline_estimator.IPUPipelineEstimator method) export_single_step() (in module tensorflow.python.ipu.serving)

F

fast (tensorflow.python.ipu.config.optimizations.optimizations.math attribute) fifo() (in module tensorflow.python.ipu.internal_ops) fixed_width_bins() (in module tensorflow.python.ipu.statistics_ops) floating_point_behaviour (tensorflow.python.ipu.config.IPUConfig attribute) from_config() (keras.ipu.optimizers.ALSGradientAccumulationOptimizer class method) (keras.ipu.optimizers.ALSOptimizer class method) (keras.ipu.optimizers.ALSOptimizerAdam class method) (keras.ipu.optimizers.ALSOptimizerRMSProp class method) (keras.ipu.optimizers.ALSOptimizerSGD class method) (keras.ipu.optimizers.GradientAccumulationOptimizer class method)

from_dict() (tensorflow.python.ipu.config.IPUConfig method) from_json() (tensorflow.python.ipu.config.IPUConfig method) frontend_attribute() (in module tensorflow.python.ipu.scopes) FunctionalExtension (class in keras.ipu.extensions) FunctionalLayerPipelineStageAssignment (class in keras.ipu) FunctionalNestedModelPipelineStageAssignment (class in keras.ipu)

G

gcl_poplar_options (tensorflow.python.ipu.config.IPUConfig attribute) gelu() (in module tensorflow.python.ipu.nn_ops) get_accumulation_factor() (tensorflow.python.ipu.embedding_ops.HostEmbeddingSGDGAOptimizerSpec method) get_attribute_metadata() (tensorflow.python.ipu.config.IPUConfig method) get_config() (ipu_tensorflow_addons.keras.layers.AssumeEqualAcrossReplicas method) (ipu_tensorflow_addons.keras.layers.CTCInferenceLayer method) (ipu_tensorflow_addons.keras.layers.CTCLoss method) (ipu_tensorflow_addons.keras.layers.CTCPredictionsLayer method) (ipu_tensorflow_addons.keras.layers.Dropout method) (ipu_tensorflow_addons.keras.layers.EffectiveTransformer method) (ipu_tensorflow_addons.keras.layers.Embedding method) (ipu_tensorflow_addons.keras.layers.GroupNormalization method) (ipu_tensorflow_addons.keras.layers.InstanceNormalization method) (ipu_tensorflow_addons.keras.layers.LayerNormalization method) (ipu_tensorflow_addons.keras.layers.PopnnGRU method) (ipu_tensorflow_addons.keras.layers.PopnnLSTM method) (ipu_tensorflow_addons.keras.layers.RecomputationCheckpoint method) (ipu_tensorflow_addons.keras.layers.SerialDense method) (ipu_tensorflow_addons.keras.optimizers.AdamIpuOptimizer method) (ipu_tensorflow_addons.keras.optimizers.LAMBIpuOptimizer method) (ipu_tensorflow_addons.keras.optimizers.SGDIpuOptimizer method) (keras.ipu.optimizers.ALSOptimizer method) (keras.ipu.optimizers.ALSOptimizerAdam method) (keras.ipu.optimizers.ALSOptimizerRMSProp method) (keras.ipu.optimizers.ALSOptimizerSGD method) (keras.ipu.optimizers.GradientAccumulationOptimizer method) get_current_iteration_counter() (in module tensorflow.python.ipu.internal_ops) get_embedding_tensor() (tensorflow.python.ipu.embedding_ops.HostEmbedding method) get_gradients() (keras.ipu.optimizers.ALSGradientAccumulationOptimizer method) (keras.ipu.optimizers.ALSOptimizer method) get_ipu_config() (in module tensorflow.python.ipu.config)

get_learning_rate() (tensorflow.python.ipu.embedding_ops.HostEmbeddingOptimizerSpec method) get_name() (tensorflow.python.ipu.optimizers.IpuOptimizer method) get_next() (tensorflow.python.ipu.ipu_infeed_queue.IPUInfeedQueue method) (tensorflow.python.ipu.ipu_infeed_queue.IPUIterator method) get_next_as_optional() (tensorflow.python.ipu.ipu_infeed_queue.IPUIterator method) get_num_of_ipus_in_device() (in module tensorflow.python.ipu.utils) get_pipeline_stage_assignment() (keras.ipu.extensions.FunctionalExtension method) (keras.ipu.extensions.ModelExtension method) (keras.ipu.extensions.SequentialExtension method) get_scaled_loss() (keras.ipu.optimizers.ALSOptimizer method) get_shard_from_colocation() (in module tensorflow.python.ipu.sharding) get_sharding() (in module tensorflow.python.ipu.sharding) get_slot() (tensorflow.python.ipu.optimizers.IpuOptimizer method) get_slot_dtype() (ipu_tensorflow_addons.keras.optimizers.AdamIpuOptimizer method) (ipu_tensorflow_addons.keras.optimizers.LAMBIpuOptimizer method) (ipu_tensorflow_addons.keras.optimizers.SGDIpuOptimizer method) get_slot_names() (tensorflow.python.ipu.optimizers.IpuOptimizer method) get_variable_names() (tensorflow.python.ipu.ipu_estimator.IPUEstimator method) (tensorflow.python.ipu.ipu_pipeline_estimator.IPUPipelineEstimator method) get_variable_value() (tensorflow.python.ipu.ipu_estimator.IPUEstimator method) (tensorflow.python.ipu.ipu_pipeline_estimator.IPUPipelineEstimator method) gradient_accumulation_count (tensorflow.python.ipu.ipu_pipeline_estimator.IPUPipelineEstimatorSpec attribute) GradientAccumulationOptimizer (class in keras.ipu.optimizers) (class in tensorflow.python.ipu.optimizers) GradientAccumulationOptimizerV2 (class in tensorflow.python.ipu.optimizers) GradientAccumulationReductionMethod (class in tensorflow.python.ipu.gradient_accumulation) (class in tensorflow.python.ipu.optimizers) group_norm() (in module tensorflow.python.ipu.normalization_ops) Grouped (tensorflow.python.ipu.pipelining_ops.PipelineSchedule attribute) GroupNorm (in module ipu_tensorflow_addons.keras.layers) GroupNormalization (class in ipu_tensorflow_addons.keras.layers) GRU (in module ipu_tensorflow_addons.keras.layers)

H

hard_sigmoid() (in module tensorflow.python.ipu.nn_ops) has_attr() (in module tensorflow.python.ipu.sharding) histogram() (in module tensorflow.python.ipu.statistics_ops) histogram_normalize() (in module tensorflow.python.ipu.statistics_ops)

histogram_update() (in module tensorflow.python.ipu.statistics_ops) host_call (tensorflow.python.ipu.ipu_estimator.IPUEstimatorSpec attribute) HostEmbedding (class in tensorflow.python.ipu.embedding_ops) HostEmbeddingOptimizerSpec (class in tensorflow.python.ipu.embedding_ops) HostEmbeddingSGDGAOptimizerSpec (class in tensorflow.python.ipu.embedding_ops)

I

inbound_layers (keras.ipu.ModelLayerPipelineStageAssignment property) (keras.ipu.NestedModelPipelineStageAssignment property) index() (tensorflow.python.ipu.ipu_estimator.IPUEstimatorSpec method) (tensorflow.python.ipu.ipu_pipeline_estimator.IPUPipelineEstimatorSpec method) infeed_benchmark() (in module tensorflow.python.ipu.dataset_benchmark) initializer (tensorflow.python.ipu.ipu_infeed_queue.IPUInfeedQueue property) instance_norm() (in module tensorflow.python.ipu.normalization_ops) InstanceNorm (in module ipu_tensorflow_addons.keras.layers) InstanceNormalization (class in ipu_tensorflow_addons.keras.layers) Interleaved (tensorflow.python.ipu.pipelining_ops.PipelineSchedule attribute) inv (tensorflow.python.ipu.config.IPUConfig.floating_point_behaviour attribute) io_tiles (tensorflow.python.ipu.config.IPUConfig attribute) ipu_jit_scope() (in module tensorflow.python.ipu.scopes) ipu_model (tensorflow.python.ipu.config.IPUConfig attribute) ipu_scope() (in module tensorflow.python.ipu.scopes) ipu_shard() (in module tensorflow.python.ipu.scopes) ipu_tensorflow_addons.keras.layers module ipu_tensorflow_addons.keras.optimizers module ipu_tensorflow_addons.v1.layers module

ipu_tensorflow_addons.v1.optimizers module IPUConfig (class in tensorflow.python.ipu.config) IPUEstimator (class in tensorflow.python.ipu.ipu_estimator) IPUEstimatorSpec (class in tensorflow.python.ipu.ipu_estimator) IPUExtendedV1 (class in tensorflow.python.ipu.ipu_strategy) IPUInfeedQueue (class in tensorflow.python.ipu.ipu_infeed_queue) IPUIterator (class in tensorflow.python.ipu.ipu_infeed_queue) IPULoggingTensorHook (class in tensorflow.python.ipu.ipu_session_run_hooks) IpuOptimizer (class in tensorflow.python.ipu.optimizers) IPUOutfeedMode (class in tensorflow.python.ipu.ipu_outfeed_queue) IPUOutfeedQueue (class in tensorflow.python.ipu.ipu_outfeed_queue) IPUOutfeedQueueIterator (class in tensorflow.python.ipu.ipu_outfeed_queue) IPUOwnedIterator (class in tensorflow.python.ipu.ipu_infeed_queue) IPUPipelineEstimator (class in tensorflow.python.ipu.ipu_pipeline_estimator) IPUPipelineEstimatorSpec (class in tensorflow.python.ipu.ipu_pipeline_estimator) IPURunConfig (class in tensorflow.python.ipu.ipu_run_config) IPUStrategy (in module tensorflow.python.ipu.ipu_strategy) IPUStrategyV1 (class in tensorflow.python.ipu.ipu_strategy) is_nested_model (keras.ipu.ModelLayerPipelineStageAssignment property) (keras.ipu.NestedModelPipelineStageAssignment property) (keras.ipu.SequentialLayerPipelineStageAssignment property) (keras.ipu.SequentialNestedModelPipelineStageAssignment property)

K

keras.ipu module keras.ipu.extensions module

keras.ipu.optimizers module

L

LAMBIpuOptimizer (class in ipu_tensorflow_addons.keras.optimizers) latest_checkpoint() (tensorflow.python.ipu.ipu_estimator.IPUEstimator method) (tensorflow.python.ipu.ipu_pipeline_estimator.IPUPipelineEstimator method) layer (keras.ipu.ModelLayerPipelineStageAssignment property) (keras.ipu.SequentialLayerPipelineStageAssignment property) layer_norm() (in module tensorflow.python.ipu.normalization_ops) LayerNorm (in module ipu_tensorflow_addons.keras.layers)

LayerNormalization (class in ipu_tensorflow_addons.keras.layers) log() (tensorflow.python.ipu.ipu_session_run_hooks.IPULoggingTensorHook method) LoggingMode (tensorflow.python.ipu.ipu_session_run_hooks.IPULoggingTensorHook attribute) lookup() (tensorflow.python.ipu.embedding_ops.HostEmbedding method) loss (tensorflow.python.ipu.ipu_estimator.IPUEstimatorSpec attribute) loss_accumulator_dtype (tensorflow.python.ipu.ipu_pipeline_estimator.IPUPipelineEstimatorSpec attribute) LSTM (in module ipu_tensorflow_addons.keras.layers)

M

MapGradientOptimizer (class in tensorflow.python.ipu.optimizers) math (tensorflow.python.ipu.config.IPUConfig.optimizations attribute) matmuls (tensorflow.python.ipu.config.IPUConfig attribute) maximum_all_gather_buffer_size (tensorflow.python.ipu.config.IPUConfig.optimizations attribute) maximum_cross_replica_sum_buffer_size (tensorflow.python.ipu.config.IPUConfig.optimizations attribute) maximum_inter_ipu_copies_buffer_size (tensorflow.python.ipu.config.IPUConfig.optimizations attribute) maximum_reduce_many_buffer_size (tensorflow.python.ipu.config.IPUConfig.optimizations attribute) maximum_reduce_scatter_buffer_size (tensorflow.python.ipu.config.IPUConfig.optimizations attribute) maximum_scheduler_lookahead_depth (tensorflow.python.ipu.config.IPUConfig.scheduling attribute) maximum_scheduler_search_space_size (tensorflow.python.ipu.config.IPUConfig.scheduling attribute) maximum_send_recv_cluster_size (tensorflow.python.ipu.config.IPUConfig.optimizations attribute) merge_infeed_io_copies (tensorflow.python.ipu.config.IPUConfig.optimizations attribute) merge_remote_buffers (tensorflow.python.ipu.config.IPUConfig.optimizations attribute) MergeRemoteBuffersBehaviour (class in tensorflow.python.ipu.config) minimum_remote_tensor_size (tensorflow.python.ipu.config.IPUConfig.optimizations attribute) mode (tensorflow.python.ipu.ipu_estimator.IPUEstimatorSpec attribute) (tensorflow.python.ipu.ipu_pipeline_estimator.IPUPipelineEstimatorSpec attribute) model_fn (tensorflow.python.ipu.ipu_estimator.IPUEstimator property) (tensorflow.python.ipu.ipu_pipeline_estimator.IPUPipelineEstimator property) ModelExtension (class in keras.ipu.extensions) ModelLayerPipelineStageAssignment (class in keras.ipu) module ipu_tensorflow_addons.keras.layers ipu_tensorflow_addons.keras.optimizers ipu_tensorflow_addons.v1.layers ipu_tensorflow_addons.v1.optimizers keras.ipu keras.ipu.extensions keras.ipu.optimizers tensorflow.python.ipu tensorflow.python.ipu.all_to_all_op tensorflow.python.ipu.application_compile_op tensorflow.python.ipu.config tensorflow.python.ipu.control_flow_ops tensorflow.python.ipu.cross_replica_ops tensorflow.python.ipu.custom_ops tensorflow.python.ipu.data.ops.dataset_ops tensorflow.python.ipu.dataset_benchmark tensorflow.python.ipu.distributed tensorflow.python.ipu.distributed.popdist_strategy tensorflow.python.ipu.embedded_runtime tensorflow.python.ipu.embedding_ops tensorflow.python.ipu.functional_ops tensorflow.python.ipu.gradient_accumulation tensorflow.python.ipu.image_ops tensorflow.python.ipu.internal_ops tensorflow.python.ipu.ipu_compiler tensorflow.python.ipu.ipu_estimator tensorflow.python.ipu.ipu_infeed_queue tensorflow.python.ipu.ipu_outfeed_queue tensorflow.python.ipu.ipu_pipeline_estimator tensorflow.python.ipu.ipu_run_config tensorflow.python.ipu.ipu_session_run_hooks tensorflow.python.ipu.ipu_strategy tensorflow.python.ipu.loops tensorflow.python.ipu.math_ops tensorflow.python.ipu.nn_ops tensorflow.python.ipu.normalization_ops tensorflow.python.ipu.optimizers tensorflow.python.ipu.pipelining_ops tensorflow.python.ipu.rand_ops tensorflow.python.ipu.reduce_scatter_op tensorflow.python.ipu.replication_ops tensorflow.python.ipu.scopes tensorflow.python.ipu.serving tensorflow.python.ipu.sharding tensorflow.python.ipu.slicing_ops tensorflow.python.ipu.statistics_ops tensorflow.python.ipu.utils tensorflow.python.ipu.within_replica_ops

move_variable_initialization_to_cpu() (in module tensorflow.python.ipu.utils) multi_conv() (in module tensorflow.python.ipu.nn_ops) multi_replica_distribution (tensorflow.python.ipu.config.IPUConfig.experimental attribute)

N

name (tensorflow.python.ipu.gradient_accumulation.Enum attribute) name() (tensorflow.python.ipu.embedded_runtime.RuntimeContext method) nanoo (tensorflow.python.ipu.config.IPUConfig.floating_point_behaviour attribute) nce_loss() (in module tensorflow.python.ipu.nn_ops) nested_model (keras.ipu.NestedModelPipelineStageAssignment property) (keras.ipu.SequentialNestedModelPipelineStageAssignment property) NestedModelPipelineStageAssignment (class in keras.ipu)

node_index (keras.ipu.ModelLayerPipelineStageAssignment property) (keras.ipu.NestedModelPipelineStageAssignment property) non_slot_devices() (tensorflow.python.ipu.ipu_strategy.IPUExtendedV1 method) normalise_image() (in module tensorflow.python.ipu.image_ops) norms (tensorflow.python.ipu.config.IPUConfig attribute) num_io_tiles (tensorflow.python.ipu.config.IPUConfig.io_tiles attribute) number_of_tuple_elements (tensorflow.python.ipu.ipu_infeed_queue.IPUInfeedQueue property)

O

oflo (tensorflow.python.ipu.config.IPUConfig.floating_point_behaviour attribute) optimizations (tensorflow.python.ipu.config.IPUConfig attribute) optimizer_function (tensorflow.python.ipu.ipu_pipeline_estimator.IPUPipelineEstimatorSpec attribute) OptimizerFunctionOutput (class in tensorflow.python.ipu.pipelining_ops)

outlined_function() (in module tensorflow.python.ipu) (in module tensorflow.python.ipu.functional_ops) output_types() (tensorflow.python.ipu.embedded_runtime.RuntimeContext method) outside_compilation_scope() (in module tensorflow.python.ipu.scopes)

P

parameter_devices (tensorflow.python.ipu.ipu_strategy.IPUExtendedV1 property) partials_type() (in module tensorflow.python.ipu.scopes) pipeline() (in module tensorflow.python.ipu.pipelining_ops) pipeline_op_kwargs (tensorflow.python.ipu.ipu_pipeline_estimator.IPUPipelineEstimatorSpec attribute) pipeline_stage (keras.ipu.ModelLayerPipelineStageAssignment property) (keras.ipu.SequentialLayerPipelineStageAssignment property) pipeline_stage_assignments (keras.ipu.NestedModelPipelineStageAssignment property) (keras.ipu.SequentialNestedModelPipelineStageAssignment property) PipelineSchedule (class in tensorflow.python.ipu.pipelining_ops) PipelineStage (class in keras.ipu) PipelineStageOptions (class in tensorflow.python.ipu.pipelining_ops) place_ops_on_io_tiles (tensorflow.python.ipu.config.IPUConfig.io_tiles attribute) pooling (tensorflow.python.ipu.config.IPUConfig attribute) PopDistStrategy (class in tensorflow.python.ipu.distributed.popdist_strategy) poplar_options (tensorflow.python.ipu.config.IPUConfig.convolutions attribute) (tensorflow.python.ipu.config.IPUConfig.matmuls attribute) (tensorflow.python.ipu.config.IPUConfig.pooling attribute) (tensorflow.python.ipu.config.IPUConfig.slices attribute) PopnnAUGRU (class in ipu_tensorflow_addons.v1.layers)

PopnnDynamicGRU (class in ipu_tensorflow_addons.v1.layers) PopnnDynamicLSTM (class in ipu_tensorflow_addons.v1.layers) PopnnGRU (class in ipu_tensorflow_addons.keras.layers) (class in ipu_tensorflow_addons.v1.layers) PopnnLSTM (class in ipu_tensorflow_addons.keras.layers) (class in ipu_tensorflow_addons.v1.layers) precompiled_user_op() (in module tensorflow.python.ipu.custom_ops) predict() (tensorflow.python.ipu.ipu_estimator.IPUEstimator method) (tensorflow.python.ipu.ipu_pipeline_estimator.IPUPipelineEstimator method) prediction_hooks (tensorflow.python.ipu.ipu_estimator.IPUEstimatorSpec attribute) (tensorflow.python.ipu.ipu_pipeline_estimator.IPUPipelineEstimatorSpec attribute) predictions (tensorflow.python.ipu.ipu_estimator.IPUEstimatorSpec attribute) prefetch_data_streams (tensorflow.python.ipu.config.IPUConfig.optimizations attribute) print_pipeline_stage_assignment_summary() (keras.ipu.extensions.FunctionalExtension method) (keras.ipu.extensions.ModelExtension method) (keras.ipu.extensions.SequentialExtension method) print_tensor() (in module tensorflow.python.ipu.internal_ops) process_count (tensorflow.python.ipu.config.IPUConfig.experimental.multi_replica_distribution attribute) process_index (tensorflow.python.ipu.config.experimental.experimental.multi_replica_distribution attribute) propagate_sharding() (in module tensorflow.python.ipu.sharding)

R

recomputation_checkpoint() (in module tensorflow.python.ipu.pipelining_ops) RecomputationCheckpoint (class in ipu_tensorflow_addons.keras.layers) RecomputationMode (class in tensorflow.python.ipu.pipelining_ops) reduce() (in module tensorflow.python.ipu.pipelining_ops) reduce_scatter() (in module tensorflow.python.ipu.reduce_scatter_op) (in module tensorflow.python.ipu.within_replica_ops) reduction_method (tensorflow.python.ipu.ipu_pipeline_estimator.IPUPipelineEstimatorSpec attribute) register() (tensorflow.python.ipu.embedding_ops.HostEmbedding method) remap() (in module tensorflow.python.ipu.internal_ops) remap_deduce() (in module tensorflow.python.ipu.internal_ops) repeat() (in module tensorflow.python.ipu.loops) ReplicatedMetricReductionMethod (class in keras.ipu) (class in keras.ipu.extensions)

replication_index() (in module tensorflow.python.ipu.replication_ops) reset() (keras.ipu.optimizers.ALSOptimizer method) reset_ipu_configuration() (in module tensorflow.python.ipu.config) reset_ipu_seed() (in module tensorflow.python.ipu.utils) reset_pipeline_stage_assignment() (keras.ipu.extensions.FunctionalExtension method) (keras.ipu.extensions.ModelExtension method) (keras.ipu.extensions.SequentialExtension method) run() (tensorflow.python.ipu.ipu_strategy.IPUStrategyV1 method) RunConfig (class in tensorflow.python.ipu.ipu_run_config) running_on_ipu_model() (in module tensorflow.python.ipu.utils) RuntimeContext (class in tensorflow.python.ipu.embedded_runtime)

S

sampled_softmax_loss() (in module tensorflow.python.ipu.nn_ops) scheduling (tensorflow.python.ipu.config.IPUConfig attribute) SchedulingAlgorithm (class in tensorflow.python.ipu.config) ScopedIPUOutfeedQueue (class in tensorflow.python.ipu.ipu_outfeed_queue) segment_sum() (in module tensorflow.python.ipu.math_ops) select_ipus (tensorflow.python.ipu.config.IPUConfig attribute) selection_order (tensorflow.python.ipu.config.IPUConfig attribute) SelectionOrder (class in tensorflow.python.ipu.config) sequence_slice() (in module tensorflow.python.ipu.slicing_ops) sequence_slice_pack() (in module tensorflow.python.ipu.slicing_ops) sequence_slice_unpack() (in module tensorflow.python.ipu.slicing_ops) Sequential (tensorflow.python.ipu.pipelining_ops.PipelineSchedule attribute) SequentialExtension (class in keras.ipu.extensions) SequentialLayerPipelineStageAssignment (class in keras.ipu) SequentialNestedModelPipelineStageAssignment (class in keras.ipu) SerialDense (class in ipu_tensorflow_addons.keras.layers) serialization_output_folder (tensorflow.python.ipu.config.IPUConfig attribute) serialized_matmul() (in module tensorflow.python.ipu.math_ops) set_all (tensorflow.python.ipu.config.IPUConfig.floating_point_behaviour attribute) set_asynchronous_callbacks() (keras.ipu.extensions.FunctionalExtension method) (keras.ipu.extensions.ModelExtension method) (keras.ipu.extensions.SequentialExtension method) set_gradient_accumulation_options() (keras.ipu.extensions.FunctionalExtension method) (keras.ipu.extensions.ModelExtension method) (keras.ipu.extensions.SequentialExtension method) set_infeed_queue_options() (keras.ipu.extensions.FunctionalExtension method) (keras.ipu.extensions.ModelExtension method) (keras.ipu.extensions.SequentialExtension method)

set_outfeed_queue_options() (keras.ipu.extensions.FunctionalExtension method) (keras.ipu.extensions.ModelExtension method) (keras.ipu.extensions.SequentialExtension method) set_pipeline_stage_assignment() (keras.ipu.extensions.FunctionalExtension method) (keras.ipu.extensions.ModelExtension method) (keras.ipu.extensions.SequentialExtension method) set_pipelining_options() (keras.ipu.extensions.FunctionalExtension method) (keras.ipu.extensions.ModelExtension method) (keras.ipu.extensions.SequentialExtension method) set_replication_options() (keras.ipu.extensions.FunctionalExtension method) (keras.ipu.extensions.ModelExtension method) (keras.ipu.extensions.SequentialExtension method) SGDIpuOptimizer (class in ipu_tensorflow_addons.keras.optimizers) ShardedOptimizer (class in tensorflow.python.ipu.optimizers) signature() (tensorflow.python.ipu.embedded_runtime.RuntimeContext method) slices (tensorflow.python.ipu.config.IPUConfig attribute) softmax() (in module tensorflow.python.ipu.nn_ops) start_output() (tensorflow.python.ipu.embedded_runtime.RuntimeContext method) state_shape() (ipu_tensorflow_addons.keras.layers.PopnnGRU method) (ipu_tensorflow_addons.keras.layers.PopnnLSTM method) (ipu_tensorflow_addons.v1.layers.PopnnGRU method) (ipu_tensorflow_addons.v1.layers.PopnnLSTM method) stochastic_rounding() (in module tensorflow.python.ipu.scopes) StochasticRoundingBehaviour (class in tensorflow.python.ipu.config) swish() (in module tensorflow.python.ipu.nn_ops)

T

tensorflow.python.ipu module tensorflow.python.ipu.all_to_all_op module tensorflow.python.ipu.application_compile_op module tensorflow.python.ipu.config module tensorflow.python.ipu.control_flow_ops module tensorflow.python.ipu.cross_replica_ops module tensorflow.python.ipu.custom_ops module tensorflow.python.ipu.data.ops.dataset_ops module tensorflow.python.ipu.dataset_benchmark module tensorflow.python.ipu.distributed module tensorflow.python.ipu.distributed.popdist_strategy module tensorflow.python.ipu.embedded_runtime module tensorflow.python.ipu.embedding_ops module tensorflow.python.ipu.functional_ops module tensorflow.python.ipu.gradient_accumulation module tensorflow.python.ipu.image_ops module tensorflow.python.ipu.internal_ops module tensorflow.python.ipu.ipu_compiler module tensorflow.python.ipu.ipu_estimator module tensorflow.python.ipu.ipu_infeed_queue module tensorflow.python.ipu.ipu_outfeed_queue module tensorflow.python.ipu.ipu_pipeline_estimator module tensorflow.python.ipu.ipu_run_config module

tensorflow.python.ipu.ipu_session_run_hooks module tensorflow.python.ipu.ipu_strategy module tensorflow.python.ipu.loops module tensorflow.python.ipu.math_ops module tensorflow.python.ipu.nn_ops module tensorflow.python.ipu.normalization_ops module tensorflow.python.ipu.optimizers module tensorflow.python.ipu.pipelining_ops module tensorflow.python.ipu.rand_ops module tensorflow.python.ipu.reduce_scatter_op module tensorflow.python.ipu.replication_ops module tensorflow.python.ipu.scopes module tensorflow.python.ipu.serving module tensorflow.python.ipu.sharding module tensorflow.python.ipu.slicing_ops module tensorflow.python.ipu.statistics_ops module tensorflow.python.ipu.utils module tensorflow.python.ipu.within_replica_ops module tiles_per_ipu (tensorflow.python.ipu.config.IPUConfig.ipu_model attribute) to_dict() (tensorflow.python.ipu.config.IPUConfig method) to_json() (tensorflow.python.ipu.config.IPUConfig method) train() (tensorflow.python.ipu.ipu_estimator.IPUEstimator method) (tensorflow.python.ipu.ipu_pipeline_estimator.IPUPipelineEstimator method) train_op (tensorflow.python.ipu.ipu_estimator.IPUEstimatorSpec attribute) training_hooks (tensorflow.python.ipu.ipu_estimator.IPUEstimatorSpec attribute) (tensorflow.python.ipu.ipu_pipeline_estimator.IPUPipelineEstimatorSpec attribute) triangular_solve_expander_block_size (tensorflow.python.ipu.config.IPUConfig.optimizations attribute) type (tensorflow.python.ipu.config.IPUConfig.device_connection attribute)

U

update_ipu_config() (tensorflow.python.ipu.distributed.popdist_strategy.PopDistStrategy method)

use_stable_statistics (tensorflow.python.ipu.config.IPUConfig.norms attribute) use_synthetic_data_for() (in module tensorflow.python.ipu.utils)

V

value (tensorflow.python.ipu.gradient_accumulation.Enum attribute) value_container() (tensorflow.python.ipu.ipu_strategy.IPUExtendedV1 method)

variables() (tensorflow.python.ipu.optimizers.IpuOptimizer method) version (tensorflow.python.ipu.config.IPUConfig.device_connection attribute) (tensorflow.python.ipu.config.IPUConfig.ipu_model attribute)

W

while_loop() (in module tensorflow.python.ipu.loops)

worker_devices (tensorflow.python.ipu.ipu_strategy.IPUExtendedV1 property)