MCPcopy Create free account

hub / github.com/0b01/tensorscript / functions

Functions219 in github.com/0b01/tensorscript

⨍ Functions 219◇ Types & classes 73
↓ 45 callersMethodcollect
(&mut self, typed_term: &TyTerm)
trsc/src/typing/constraint.rs:42
↓ 43 callersMethodas_str
(&self)
trsc/src/typing/type_env.rs:30
↓ 32 callersMethodadd
(&mut self, a: Type, b: Type)
trsc/src/typing/constraint.rs:37
↓ 28 callersMethodresolve_type
resolve the type of an identifier first check current mod name, if it doesn not exist, then check in the global scope
trsc/src/typing/type_env.rs:155
↓ 23 callersMethodconvert_span
(&self, sp: &PestSpan)
trsc/src/span.rs:22
↓ 16 callersMethodapply_ty
(&mut self, ty: &Type)
trsc/src/typing/unifier.rs:228
↓ 15 callersMethodfresh_var
create new type variable
trsc/src/typing/type_env.rs:122
↓ 14 callersMethodas_num
(&self)
trsc/src/typing/types.rs:244
↓ 12 callersMethodmodule
get current module name
trsc/src/typing/type_env.rs:391
↓ 12 callersMethodspan
(&self)
trsc/src/typing/types.rs:129
↓ 11 callersMethodadd_type
add type alias in current scope
trsc/src/typing/type_env.rs:194
↓ 11 callersMethodconsume
(&self, pair: Pair<Rule>)
trsc/src/parsing/ast_builder.rs:80
↓ 10 callersMethodprint_errs
(&self)
trsc/src/errors/emitter.rs:28
↓ 10 callersMethodto_btreemap
(&self)
trsc/src/typing/typed_term.rs:239
↓ 10 callersMethodwith_span
modifies the span parameter in type to the most relevant
trsc/src/typing/types.rs:206
↓ 9 callersMethodcollect
(&mut self, term: &TyTerm)
trsc/src/codegen/pytorch.rs:390
↓ 9 callersMethodset_module
set current module name
trsc/src/typing/type_env.rs:396
↓ 8 callersMethodindent
(&mut self)
trsc/src/codegen/pytorch.rs:58
↓ 8 callersMethodto_ty
(&self, span: &ByteSpan)
trsc/src/typing/typed_term.rs:225
↓ 8 callersMethodunify
(&mut self, cs: Constraints)
trsc/src/typing/unifier.rs:26
↓ 6 callersMethodas_vec
(&self)
trsc/src/typing/types.rs:153
↓ 6 callersMethodcond
(self, i, _)
misc/tf_fizzbuzz.py:24
↓ 6 callersMethodty
(&self)
trsc/src/typing/typed_term.rs:50
↓ 5 callersMethodannotate
(&self, term: &Term)
trsc/src/typing/annotate.rs:29
↓ 5 callersMethodas_string
(&self)
trsc/src/typing/types.rs:233
↓ 5 callersMethodshift_tab
(&mut self)
trsc/src/codegen/pytorch.rs:67
↓ 5 callersMethodtab
(&mut self)
trsc/src/codegen/pytorch.rs:63
↓ 4 callersMethodadd
(&mut self, e: Diag)
trsc/src/errors/emitter.rs:24
↓ 4 callersMethodannotate_fn_ty_sig
(&self, modname: String, name: String, sig: &FnTySig, span: &ByteSpan)
trsc/src/typing/annotate.rs:318
↓ 4 callersMethodbuild_fn_app
(&self, pair: Pair<Rule>)
trsc/src/parsing/ast_builder.rs:275
↓ 4 callersMethodbuild_fn_ty_sig
(&self, pair: Pair<Rule>)
trsc/src/parsing/ast_builder.rs:552
↓ 4 callersMethodcreate_tensor
make a new tensor based on type signature
trsc/src/typing/type_env.rs:303
↓ 4 callersMethodfind
(&self, path_name: &str, mod_name: &str)
trsc/src/core/mod.rs:84
↓ 4 callersMethodis_empty
(&self)
trsc/src/typing/constraint.rs:33
↓ 4 callersMethodpop_scope
exiting block during tree traversal
trsc/src/typing/type_env.rs:141
↓ 4 callersFunctionsubs
(typed_term: &TyTerm, s: &mut Substitution)
trsc/src/typing/inferred_ast.rs:7
↓ 4 callersMethodunify_var
(&mut self, tvar: TypeId, ty: Type)
trsc/src/typing/unifier.rs:174
↓ 3 callersMethodannotate_fn_app_arg
(&self, call: &FnAppArg)
trsc/src/typing/annotate.rs:392
↓ 3 callersMethodas_args_map
(&self)
trsc/src/typing/types.rs:161
↓ 3 callersMethodbuild_dim_tsr_assign
(&self, pair: Pair<Rule>)
trsc/src/parsing/ast_builder.rs:501
↓ 3 callersMethodcollect_term
(&mut self, term: &TyTerm, var: Option<String>, is_stmt: bool)
trsc/src/codegen/pytorch.rs:104
↓ 3 callersMethodis_resolved
(&self)
trsc/src/typing/types.rs:260
↓ 3 callersMethodrun
(self)
misc/tf_fizzbuzz.py:19
↓ 2 callersMethod_process_level
(&self, curr: Pair<Rule>)
trsc/src/parsing/ast_builder.rs:364
↓ 2 callersMethodadd_dim_alias
tie an alias with a type variable dimension
trsc/src/typing/type_env.rs:264
↓ 2 callersMethodannotate_field_access
(&self, f_a: &FieldAccess)
trsc/src/typing/annotate.rs:545
↓ 2 callersMethodannotate_tensor_ty_sig
(&self, sig: &TensorTy, _span: &ByteSpan)
trsc/src/typing/annotate.rs:328
↓ 2 callersMethodas_rank
(&self)
trsc/src/typing/types.rs:252
↓ 2 callersMethodbuild_field_access
(&self, pair: Pair<Rule>)
trsc/src/parsing/ast_builder.rs:140
↓ 2 callersMethodbuild_fn_app_args
(&self, pair: Pair<Rule>)
trsc/src/parsing/ast_builder.rs:305
↓ 2 callersMethodcollect_fn_app
(&mut self, fn_app: &TyFnApp)
trsc/src/typing/constraint.rs:191
↓ 2 callersMethodexists
check if an alias exists
trsc/src/typing/type_env.rs:350
↓ 2 callersMethodextend_arg
(&mut self, arg: &TyFnAppArg)
trsc/src/typing/typed_term.rs:201
↓ 2 callersMethodfind_mod
(&self, mod_name:&str)
trsc/src/core/mod.rs:89
↓ 2 callersMethodfirst_arg_ty
returns the first argument type of a function argument
trsc/src/typing/types.rs:193
↓ 2 callersMethodfresh_dim
create new dimension type variable
trsc/src/typing/type_env.rs:116
↓ 2 callersMethodgen_fn_app
(&self, name: &str, args: &[TyFnAppArg])
trsc/src/core/lin.rs:95
↓ 2 callersMethodgenerate
(&mut self)
trsc/src/codegen/pytorch.rs:81
↓ 2 callersMethodgenerate_fn
(&mut self)
trsc/src/codegen/pytorch.rs:132
↓ 2 callersMethodgenerate_fn_decl_head
(&mut self, name: &str, func: &TyFnDecl)
trsc/src/codegen/pytorch.rs:287
↓ 2 callersMethodget_scoped_types
iterate over scopes and find the alias in each
trsc/src/typing/type_env.rs:167
↓ 2 callersFunctionget_str_attr
(keyname: &'static str, attrs: &'a [syn::Attribute])
trsc_core_derive/src/attrs.rs:38
↓ 2 callersMethodimport_module
import module type and associated methods into type environment
trsc/src/typing/type_env.rs:401
↓ 2 callersMethodimport_node_assign
create aliases for an untyped AST node assign
trsc/src/typing/type_env.rs:356
↓ 2 callersMethodimport_top_level_ty_sig
(&mut self, mod_name: &ModName, ty_sig: &TensorTy)
trsc/src/typing/type_env.rs:376
↓ 2 callersFunctionparse
(path: &str, name: &str, toks: &[Token])
trsc_core_derive/src/parser.rs:74
↓ 2 callersMethodpush_scope
push scope onto stack during tree traversal
trsc/src/typing/type_env.rs:128
↓ 2 callersMethodpush_scope_collection
during constraint collection, push the popped scopes back
trsc/src/typing/type_env.rs:134
↓ 2 callersMethodresolve_tensor
generate a tensor from untyped ast tensor signature
trsc/src/typing/type_env.rs:335
↓ 2 callersMethodresolve_unresolved
( &mut self, ty: &Type, fn_name: &str, arg_ty: Type, ret_ty: Type,
trsc/src/typing/type_env.rs:426
↓ 2 callersFunctionsubs_fn_app_arg
(a: &TyFnAppArg, s: &mut Substitution)
trsc/src/typing/inferred_ast.rs:112
↓ 2 callersFunctionunexpected_token
(pair: &Pair<Rule>)
trsc/src/parsing/ast_builder.rs:602
↓ 2 callersFunctionunwrap_dim
(in_dim: &TyTerm)
trsc/src/core/lin.rs:122
↓ 1 callersMethod__init__
(self)
trsc/tests/output/gan.py:27
↓ 1 callersMethodadd_init
add stateful initialization in current scope
trsc/src/typing/type_env.rs:254
↓ 1 callersMethodadd_resolved_dim_alias
tie an alias with a resolved dimension
trsc/src/typing/type_env.rs:270
↓ 1 callersMethodadd_tsr_alias
tie an alias with a tensor
trsc/src/typing/type_env.rs:282
↓ 1 callersMethodadd_type_allow_replace
add type alias in current scope(allows replacement)
trsc/src/typing/type_env.rs:221
↓ 1 callersMethodannotate_decl
(&self, decl: &Decl)
trsc/src/typing/annotate.rs:199
↓ 1 callersMethodannotate_fn_app
(&self, fn_app: &FnApp)
trsc/src/typing/annotate.rs:402
↓ 1 callersMethodannotate_fn_decl
(&self, f: &FnDecl)
trsc/src/typing/annotate.rs:421
↓ 1 callersMethodannotate_fn_decl_param
(&self, p: &FnDeclParam)
trsc/src/typing/annotate.rs:528
↓ 1 callersMethodannotate_pipes
(&self, pipes: &[Term])
trsc/src/typing/annotate.rs:115
↓ 1 callersMethodannotate_tuples
(&self, tup: &[Term], s: &ByteSpan)
trsc/src/typing/annotate.rs:104
↓ 1 callersMethodannotate_view_fn
(&self, v_fn: &ViewFn, arg: &TyFnAppArg)
trsc/src/typing/annotate.rs:185
↓ 1 callersMethodannotate_weights_assign
(&self, w_assign: &WeightsAssign)
trsc/src/typing/annotate.rs:348
↓ 1 callersMethodapply
apply substitution to a set of constraints
trsc/src/typing/unifier.rs:217
↓ 1 callersMethodas_diagnostic
(&self, code_map: &CodeMap)
trsc/src/errors/diagnostic.rs:20
↓ 1 callersMethodas_mod_name
(&self)
trsc/src/typing/types.rs:226
↓ 1 callersMethodbuild_block
(&self, pair: Pair<Rule>)
trsc/src/parsing/ast_builder.rs:111
↓ 1 callersMethodbuild_expr
(&self, pair: Pair<Rule>)
trsc/src/parsing/ast_builder.rs:160
↓ 1 callersMethodbuild_float_lit
(&self, pair: Pair<Rule>)
trsc/src/parsing/ast_builder.rs:540
↓ 1 callersMethodbuild_fn_app_arg
(&self, pair: Pair<Rule>)
trsc/src/parsing/ast_builder.rs:292
↓ 1 callersMethodbuild_fn_app_param
(&self, pair: Pair<Rule>)
trsc/src/parsing/ast_builder.rs:129
↓ 1 callersMethodbuild_fn_decl
(&self, pair: Pair<Rule>)
trsc/src/parsing/ast_builder.rs:184
↓ 1 callersMethodbuild_fn_decl_param
(&self, pair: Pair<Rule>)
trsc/src/parsing/ast_builder.rs:243
↓ 1 callersMethodbuild_fn_decl_params
(&self, pair: Pair<Rule>)
trsc/src/parsing/ast_builder.rs:261
↓ 1 callersMethodbuild_fn_decls
(&self, pair: Pair<Rule>)
trsc/src/parsing/ast_builder.rs:237
↓ 1 callersMethodbuild_graph_decl
(&self, pair: Pair<Rule>)
trsc/src/parsing/ast_builder.rs:435
↓ 1 callersMethodbuild_graph_decl_body
(&self, pair: Pair<Rule>)
trsc/src/parsing/ast_builder.rs:458
next →1–100 of 219, ranked by callers