MCPcopy Create free account

hub / github.com/OpenNMT/CTranslate2 / functions

Functions2,508 in github.com/OpenNMT/CTranslate2

↓ 2 callersMethodset_embeddings
(self, spec, module)
python/ctranslate2/converters/opennmt_tf.py:312
↓ 2 callersFunctionset_ffn
(spec, weights, scope)
python/ctranslate2/converters/marian.py:241
↓ 2 callersFunctionset_ffn
(spec, module)
python/ctranslate2/converters/fairseq.py:260
↓ 2 callersFunctionset_ffn
(spec, variables, scope)
python/ctranslate2/converters/opennmt_py.py:294
↓ 2 callersFunctionset_ffn
(spec, variables, scope)
python/ctranslate2/converters/eole_ct2.py:282
↓ 2 callersMethodset_ffn
(self, spec, module)
python/ctranslate2/converters/opennmt_tf.py:264
↓ 2 callersFunctionset_input_layers
(spec, module)
python/ctranslate2/converters/fairseq.py:236
↓ 2 callersFunctionset_input_layers
(spec, variables, scope)
python/ctranslate2/converters/opennmt_py.py:246
↓ 2 callersFunctionset_input_layers
(spec, variables, scope)
python/ctranslate2/converters/eole_ct2.py:228
↓ 2 callersFunctionset_layer_norm_auto
(spec, weights, scope)
python/ctranslate2/converters/marian.py:247
↓ 2 callersMethodset_layer_norm_from_wrapper
(self, spec, module)
python/ctranslate2/converters/opennmt_tf.py:291
↓ 2 callersFunctionset_num_threads
src/utils.cc:85
↓ 2 callersMethodset_position_encodings
(self, spec, module)
python/ctranslate2/converters/opennmt_tf.py:315
↓ 2 callersMethodset_stack
(self, spec, module, is_decoder=False)
python/ctranslate2/converters/transformers.py:1281
↓ 2 callersFunctionset_transformer_decoder
(spec, module, with_encoder_attention=True)
python/ctranslate2/converters/fairseq.py:217
↓ 2 callersFunctionset_transformer_decoder
(spec, variables, with_encoder_attention=True)
python/ctranslate2/converters/opennmt_py.py:228
↓ 2 callersFunctionset_transformer_decoder
(spec, variables, with_encoder_attention=True)
python/ctranslate2/converters/eole_ct2.py:214
↓ 2 callersMethodset_transformer_decoder
(self, spec, module, inputter)
python/ctranslate2/converters/opennmt_tf.py:235
↓ 2 callersFunctionsignbit
Check sign. See also:** Documentation for [std::signbit](https://en.cppreference.com/w/cpp/numeric/math/signbit). \param arg number to check \retval t
include/half_float/half.hpp:4399
↓ 2 callersFunctionskip_whitespace
include/nlohmann/json.hpp:8801
↓ 2 callersMethodsmooth_activation
(self, spec, activation_scales)
python/ctranslate2/converters/transformers.py:561
↓ 2 callersFunctionsort_hypotheses
Sort hypotheses from best to worst score, in the limit of max_hypotheses.
src/decoding.cc:206
↓ 2 callersMethodstart
src/thread_pool.cc:99
↓ 2 callersFunctionstarts_with
src/utils.cc:114
↓ 2 callersFunctionstring_to_bool
src/utils.cc:24
↓ 2 callersFunctionsynchronize_device
src/devices.cc:101
↓ 2 callersFunctionto_spdlog_level
src/logging.cc:12
↓ 2 callersMethodtranslate_async
src/buffered_translation_wrapper.cc:29
↓ 2 callersMethodtranslate_text_file
src/translator.cc:84
↓ 2 callersFunctionupdate_sample_with_prefix
src/decoding.cc:21
↓ 2 callersMethodupdate_state
src/layers/decoder.cc:33
↓ 2 callersMethoduse_flash_attention
include/ctranslate2/models/model.h:95
↓ 2 callersMethodwrite_bson
! @param[in] j JSON value to serialize @pre j.type() == value_t::object */
include/nlohmann/json.hpp:14975
↓ 2 callersMethodwrite_cbor
! @param[in] j JSON value to serialize */
include/nlohmann/json.hpp:15004
↓ 2 callersMethodwrite_msgpack
! @param[in] j JSON value to serialize */
include/nlohmann/json.hpp:15328
↓ 1 callersFunctionMPI_Allgather
src/cuda/mpi_stub.cc:50
↓ 1 callersFunctionMPI_Bcast
src/cuda/mpi_stub.cc:60
↓ 1 callersFunctionMPI_Comm_rank
src/cuda/mpi_stub.cc:80
↓ 1 callersFunctionMPI_Comm_size
src/cuda/mpi_stub.cc:86
↓ 1 callersFunctionMPI_Finalize
src/cuda/mpi_stub.cc:74
↓ 1 callersFunctionMPI_Init
src/cuda/mpi_stub.cc:68
↓ 1 callersMethod__init__
(self)
python/tests/test_spec.py:18
↓ 1 callersMethod__init__
Initializes the model specification. Args: num_encoder_layers: The number of encoder layers. num_encoder_heads: The numbe
python/ctranslate2/specs/whisper_spec.py:29
↓ 1 callersMethod__init__
(self)
python/ctranslate2/specs/wav2vec2bert_spec.py:53
↓ 1 callersMethod_alias_variables
Find duplicate variables in spec and create aliases.
python/ctranslate2/specs/model_spec.py:169
↓ 1 callersFunction_batch_iterator
(iterable, batch_size, batch_type)
python/ctranslate2/extensions.py:557
↓ 1 callersFunction_benchmark_translation
( image_name, source_file, target_file, environment, num_cpus, use_gpu, )
tools/benchmark/benchmark.py:135
↓ 1 callersFunction_check_generator_logits
( tmp_dir, model_name, hf_model_class, hf_tokenizer_class, input_text )
python/tests/test_transformers.py:391
↓ 1 callersFunction_count_tokens
(path)
tools/benchmark/benchmark.py:24
↓ 1 callersFunction_detokenize
(image_name, input_file, output_file)
tools/benchmark/benchmark.py:83
↓ 1 callersFunction_dtype_to_type_id
(object_dtype)
python/ctranslate2/specs/model_spec.py:280
↓ 1 callersMethod_equal
(self, other)
python/ctranslate2/specs/model_spec.py:668
↓ 1 callersFunction_flatten_vocabularies
(vocabularies)
python/ctranslate2/specs/model_spec.py:417
↓ 1 callersFunction_get_bleu_score
(hyp_file, ref_file)
tools/benchmark/benchmark.py:18
↓ 1 callersMethod_get_lang_ids_from_tokenizer
(self, tokenizer)
python/ctranslate2/converters/transformers.py:952
↓ 1 callersFunction_get_long_description
()
python/setup.py:15
↓ 1 callersFunction_get_model_config
(model)
python/ctranslate2/converters/marian.py:111
↓ 1 callersFunction_get_model_spec
(args)
python/ctranslate2/converters/fairseq.py:28
↓ 1 callersFunction_get_model_spec_lm
Creates a model specification from the model options.
python/ctranslate2/converters/opennmt_py.py:99
↓ 1 callersFunction_get_model_spec_lm
Creates a model specification from the model config.
python/ctranslate2/converters/eole_ct2.py:92
↓ 1 callersFunction_get_model_spec_seq2seq
Creates a model specification from the model options.
python/ctranslate2/converters/opennmt_py.py:58
↓ 1 callersFunction_get_model_spec_seq2seq
Creates a model specification from the model config.
python/ctranslate2/converters/eole_ct2.py:20
↓ 1 callersFunction_get_project_version
()
python/setup.py:23
↓ 1 callersMethod_load
(self)
python/ctranslate2/converters/converter.py:108
↓ 1 callersFunction_make_sinusoidal_position_encodings
(dim, num_positions=2048)
python/ctranslate2/converters/marian.py:285
↓ 1 callersFunction_maybe_add_library_root
(lib_name)
python/setup.py:31
↓ 1 callersFunction_monitor_container
(container, poll_interval=1, use_gpu=False)
tools/benchmark/benchmark.py:32
↓ 1 callersFunction_parent_scope
(scope)
python/ctranslate2/specs/model_spec.py:51
↓ 1 callersMethod_quantize
Possibly quantizes the variable of the layer.
python/ctranslate2/specs/model_spec.py:191
↓ 1 callersMethod_remove_pad_weights
(self, spec)
python/ctranslate2/converters/transformers.py:455
↓ 1 callersFunction_remove_self
(signature)
docs/conf.py:47
↓ 1 callersMethod_serialize
Serializes the model variables.
python/ctranslate2/specs/model_spec.py:382
↓ 1 callersMethod_to
(self, dtype: str)
python/ctranslate2/specs/model_spec.py:664
↓ 1 callersFunction_tokenize
(image_name, input_file, output_file)
tools/benchmark/benchmark.py:79
↓ 1 callersMethodadd
src/batch_reader.cc:123
↓ 1 callersMethodadd_scope_time
src/profiler.cc:68
↓ 1 callersMethodalign
python/cpp/whisper.cc:91
↓ 1 callersFunctionall_beams_diverged_from_prefix
src/decoding.cc:344
↓ 1 callersMethodallocator
include/ctranslate2/replica_pool.h:334
↓ 1 callersFunctionannotate_classes
(classes)
docs/generate.py:83
↓ 1 callersFunctionappend_exponent
include/nlohmann/json.hpp:17711
↓ 1 callersMethodapply
src/models/whisper.cc:213
↓ 1 callersMethodas_sequence_encoder
src/models/model.cc:157
↓ 1 callersMethodas_sequence_generator
src/models/model.cc:153
↓ 1 callersFunctionbacktrace
src/dtw.cc:8
↓ 1 callersFunctionbenchmark_conv1d
tests/benchmark_ops.cc:114
↓ 1 callersFunctionbenchmark_dequantize
tests/benchmark_ops.cc:103
↓ 1 callersFunctionbenchmark_gather
tests/benchmark_ops.cc:9
↓ 1 callersFunctionbenchmark_gemm
tests/benchmark_ops.cc:86
↓ 1 callersFunctionbenchmark_generation
(generator, sp, prompt_tokens, gene
tools/benchmark_tensor_parallel/benchmark.py:61
↓ 1 callersFunctionbenchmark_layer_norm
tests/benchmark_ops.cc:35
↓ 1 callersFunctionbenchmark_masked_softmax
tests/benchmark_ops.cc:56
↓ 1 callersFunctionbenchmark_median_filter
tests/benchmark_ops.cc:123
↓ 1 callersFunctionbenchmark_quantize
tests/benchmark_ops.cc:95
↓ 1 callersFunctionbenchmark_softmax
tests/benchmark_ops.cc:48
↓ 1 callersFunctionbenchmark_split
tests/benchmark_ops.cc:26
↓ 1 callersFunctionbenchmark_topk
tests/benchmark_ops.cc:74
↓ 1 callersFunctionbenchmark_transpose
tests/benchmark_ops.cc:19
↓ 1 callersMethodbinary_version
include/ctranslate2/models/model.h:53
↓ 1 callersFunctionbuild_alibi
src/layers/attention_layer.cc:13
← previousnext →501–600 of 2,508, ranked by callers