Searched refs:input_array_a (Results 1 – 1 of 1) sorted by relevance
39 const auto& input_array_a = model->GetArray(input_lhs); in UnrollBatchMatMul3D() local41 const int dims_count = input_array_a.shape().dimensions_count(); in UnrollBatchMatMul3D()46 std::vector<int> slice_size_a = input_array_a.shape().dims(); in UnrollBatchMatMul3D()59 slice_a_op_output.data_type = input_array_a.data_type; in UnrollBatchMatMul3D()67 {-1, input_array_a.shape().dims(dims_count - 1)})}; in UnrollBatchMatMul3D()72 slice_a_reshape_op_output.data_type = input_array_a.data_type; in UnrollBatchMatMul3D()113 matmul_op_output.data_type = input_array_a.data_type; in UnrollBatchMatMul3D()125 const auto& input_array_a = model->GetArray(input_lhs); in UnrollBatchMatMulRecursion() local126 const auto& dims_vec = input_array_a.shape().dims(); in UnrollBatchMatMulRecursion()153 pack_op_output.data_type = input_array_a.data_type; in UnrollBatchMatMulRecursion()[all …]