Home
last modified time | relevance | path

Searched refs:set_parameter_to_constants (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dtransformer.cpp14 void set_parameter_to_constants( in set_parameter_to_constants() function
49 set_parameter_to_constants<T_LAYER>(layer, tensor_options); in get_a_test_layer()
1387 set_parameter_to_constants<Transformer>(model, tensor_options); in transformer_test_helper()
1413 set_parameter_to_constants<Transformer>(model_cus, tensor_options); in transformer_test_helper()