Searched defs:tensor_options (Results 1 – 3 of 3) sorted by relevance
16 const torch::TensorOptions& tensor_options) { in set_parameter_to_constants()28 const torch::TensorOptions& tensor_options, in get_a_test_layer()59 torch::TensorOptions tensor_options = in transformer_encoder_layer_test_helper() local250 torch::TensorOptions tensor_options = in transformer_decoder_layer_test_helper() local456 torch::TensorOptions tensor_options = in transformer_decoder_layer_test_helper_gelu() local570 torch::TensorOptions tensor_options = in transformer_encoder_test_helper() local844 torch::TensorOptions tensor_options = in transformer_decoder_test_helper() local1369 torch::TensorOptions tensor_options = in transformer_test_helper() local
40 auto tensor_options = typeIdWithDefault(r, 2, dispatch_key); in nested_tensor_ctor() local
122 …auto tensor_options = at::TensorOptions().dtype(dtype).layout(layout).device(device).pinned_memory… in linspace_logspace_batch_rule_helper() local