Searched refs:mat_a_layout (Results 1 – 1 of 1) sorted by relevance
326 TensorLayout mat_a_layout, mat_b_layout, output_layout; in InferTensorLayout() local327 …if ((mat_a_layout.InitFromVector(dev_matrix_shape_, inputs_tensor_map_[0], inputs_shape_[0]) != SU… in InferTensorLayout()337 inputs_layout->push_back(mat_a_layout); in InferTensorLayout()350 TensorLayout mat_a_layout = inputs_layout.at(0); in InferTensorInfo() local353 TensorInfo mat_a_tensor_info(mat_a_layout); in InferTensorInfo()