Searched defs:LinearParams (Results 1 – 2 of 2) sorted by relevance
40 struct LinearParams { struct41 c10::DeviceIndex device_id;42 cudnnDataType_t dataType;43 int64_t input_size[max_num_input_dim];44 uint8_t input_dim;45 at::MemoryFormat memory_format;46 int64_t weight_size[2];47 bool deterministic;48 bool allow_tf32;
40 using LinearParams = ideep::matmul_forward_params; variable