MCPcopy Create free account

hub / github.com/tensorflow/lingvo / endpoints

Endpoints48 in github.com/tensorflow/lingvo

⨍ Functions 9,552◇ Types & classes 1,444↳ Endpoints 48
RouteComputeContextVectorWithAttenProbs
py_utils.NameScopeDecorator( 'MultiHeadedAttention/ComputeContextVectorWithAttenProbs' )
lingvo/core/attention.py:None
RouteComputeContextVectorWithCachedSource
py_utils.NameScopeDecorator( 'MultiHeadedAttention/ComputeContextVectorWithCachedSource' )
lingvo/core/attention.py:None
RouteComputeContextVectorWithSource
py_utils.NameScopeDecorator( 'MultiHeadedAttention/ComputeContextVectorWithSource' )
lingvo/core/attention.py:None
RouteComputePredictions
py_utils.NameScopeDecorator('MTDecoderV1/ComputePredictions')
lingvo/tasks/mt/decoder.py:None
RouteFn
tf.function( input_signature=[tf.TensorSpec(shape=None, dtype=t) for t in dtypes])
lingvo/core/py_utils_test.py:None
RouteFn
tf.function(input_signature=input_signature, autograph=False)
lingvo/core/py_utils.py:None
RouteForward
_WrapFunction(input_signature=Flatten(fwd_sig))
lingvo/core/py_utils.py:None
RouteFunc
test_utils.DefineAndTrace(x, target_rank)
lingvo/core/py_utils_test.py:None
RouteFwd
py_utils.Function(fwd_sig=[theta_specs, state0_specs, inputs_specs])
lingvo/core/recurrent.py:None
Route__call__
tf.function(input_signature=[ tf.TensorSpec(shape=[None, None, None, None], dtype=tf.float32), tf.
lingvo/tasks/milan/tf_hub_layers_test.py:None
RoutetestApplyPackingErrors
parameterized.named_parameters( ('Test1', tf.errors.InvalidArgumentError, 'out of bound', ApplyPa
lingvo/core/ops/pack_ops_test.py:None
RoutetestBasic
parameterized.named_parameters(('OutputList', False, 8), ('OutputNestedMap',
lingvo/core/generic_input_test.py:None
RoutetestBasic
parameterized.named_parameters( ('OutputListEager', False, 8, False), ('OutputNestedMapEager', Tru
lingvo/core/generic_input_tf2_test.py:None
RoutetestBatchNormLayerFProp
parameterized.named_parameters(('F32Input', tf.float32, 47.8371887), ('BF16I
lingvo/core/layers_test.py:None
RoutetestBlockSparseAttentionFprop
parameterized.named_parameters( { 'testcase_name': '_local_attention', 'expected_out
lingvo/core/self_attention_layer_test.py:None
RoutetestBothPolicies
test.mock.patch.object(saver, 'time')
lingvo/core/saver_test.py:None
RoutetestBothPoliciesPreempt
test.mock.patch.object(saver, 'time')
lingvo/core/saver_test.py:None
RoutetestClearRelativeAttentionInTransformerLayer
parameterized.named_parameters( { 'testcase_name': 'apply_residual', 'apply_residual
lingvo/core/batch_major_attention_test.py:None
RoutetestDotProductDtype
parameterized.named_parameters( # Use the default data types. ('dtype_default', [], 1e-06),
lingvo/core/batch_major_attention_test.py:None
RoutetestEinsumQuantization
parameterized.named_parameters( ( '4bit_weight_qat_output_false', False, '
lingvo/core/py_utils_test.py:None
RoutetestEinsumSoftmax
parameterized.named_parameters( {'testcase_name': 'no_label_smoothing', 'expected_loss': 27.981373},
lingvo/core/softmax_layers_test.py:None
RoutetestFCLayerDtypes
parameterized.named_parameters( ('F32FPropF32Input', tf.float32, tf.float32, 0.668211), ('F32FProp
lingvo/core/layers_test.py:None
RoutetestFPropDtypes
parameterized.named_parameters( ('F32FPropF32Input', tf.float32, tf.float32), ('F32FPropBF16Input'
lingvo/core/conformer_layer_test.py:None
RoutetestFPropDtypes
parameterized.named_parameters( ('F32FPropF32Input', tf.float32, tf.float32, 31.040909), ('F32FPro
lingvo/core/layers_test.py:None
RoutetestFPropDtypesWithListRegexDtypes
parameterized.named_parameters( ('DefaultFp32', [], tf.float32), ('RegexDtypeFp16', [(r'.*(fflayer
lingvo/core/conformer_layer_test.py:None
RoutetestGPipeBatchMajorTransformerFProp
parameterized.named_parameters( { 'testcase_name': '_one_split_one_mb_packed', 'spli
lingvo/core/layers_with_gpipe_test.py:None
RoutetestGPipeTransformerDecoderStackFPropWithEmbeddings
parameterized.named_parameters( _AddClassesToTestParams(({ 'testcase_name': '_split1',
lingvo/core/layers_with_gpipe_test.py:None
RoutetestGPipeTransformerLmModel
parameterized.named_parameters( _AddClassesToTestParams(({ 'testcase_name': '_split1',
lingvo/core/layers_with_gpipe_test.py:None
RoutetestGPipeTransformerMtModel
parameterized.named_parameters( _AddClassesToTestParams(({ 'testcase_name': '_split1',
lingvo/core/layers_with_gpipe_test.py:None
RoutetestGroupPairAUCMetricTargetSorting
parameterized.named_parameters( dict( testcase_name='all_positive_first', target=[1.
lingvo/core/metrics_test.py:None
RoutetestLSTMCellGrouped
parameterized.named_parameters(('_WithoutOutputShuffling', 1), ('_WithOutput
lingvo/core/rnn_cell_test.py:None
RoutetestLSTMCellSimple
parameterized.named_parameters( ('HasBias', True), ('NoBias', False), ('HasBiasPackedInputs'
lingvo/core/lstm_frnn_layer_test.py:None
RoutetestLSTMSimpleWithForgetGateInitBias
parameterized.named_parameters( ('', False, [0.1, 0.1, 0.1, 0.1, 0.1, 0.1, 2.1, 2.1, 2.1, 0.1, 0.
lingvo/core/rnn_cell_test.py:None
RoutetestLSTMSimple_P3
parameterized.named_parameters( dict( testcase_name='_GlobalNoise', m_expected=[[-0.
lingvo/core/rnn_cell_test.py:None
RoutetestLayerNormDtypes
parameterized.named_parameters( { 'testcase_name': 'F32Input', 'input_dtype': tf.flo
lingvo/core/layers_test.py:None
RoutetestMultiHeadedAttentionDotProduct
parameterized.named_parameters( # Use the default data types. ('dtype_default', [], 1e-06),
lingvo/core/batch_major_attention_test.py:None
RoutetestPackSingleSequence
parameterized.named_parameters( # Output with batch size 1 containing all 3 inputs. ('Ex1Sequentia
lingvo/core/ops/pack_ops_test.py:None
RoutetestProjectionLayerMeta
parameterized.named_parameters( { 'testcase_name': 'RELU', 'activation': 'RELU',
lingvo/core/layers_test.py:None
RoutetestQAT
parameterized.named_parameters( ('DefaultQatWeight', False, [[3.0, 3.0, 3.0]]), ('TestQatOutput',
lingvo/core/py_utils_test.py:None
RoutetestReZeroAddLayerLayer
parameterized.named_parameters( { 'testcase_name': 'apply_residual', 'apply_residual
lingvo/core/batch_major_attention_test.py:None
RoutetestSRUCell
parameterized.named_parameters( ('_Regular', None, True, 0, False, False, False, False, [[0.58657
lingvo/core/rnn_cell_test.py:None
RoutetestSimpleCase
parameterized.named_parameters( { 'testcase_name': 'max_pooling', 'pooling_type': 'M
lingvo/core/conv_layers_builder_test.py:None
RoutetestSimpleCases
parameterized.named_parameters( { 'testcase_name': 'max_pooling_same_padding', 'pool
lingvo/core/layers_test.py:None
RoutetestTpuHostDecorator
mock.patch.object(tf.tpu, 'outside_compilation', autospec=True)
lingvo/core/py_utils_test.py:None
RoutetestTransformerFeedForwardLayerFPropDtype
parameterized.named_parameters( ('F32FPropF32Input', tf.float32, tf.float32, 7.182965), ('F32FProp
lingvo/core/layers_with_attention_test.py:None
RoutetestTransformerLayerFPropDtypes
parameterized.named_parameters( ('F32FPropF32Input', tf.float32, tf.float32), ('F32FPropBF16Input'
lingvo/core/batch_major_attention_test.py:None
RoutetestTransformerStack
parameterized.named_parameters( {'testcase_name': '_baseline', 'num_splits': 1, 'num_micro_batches': 1},
lingvo/core/self_attention_layer_test.py:None
Routetest_normalize_trailing_eos
parameterized.named_parameters( { 'testcase_name': 'remove_eos', 'need_trailing_eos'
lingvo/tasks/asr/eos_normalization_test.py:None