Searched refs:input_array_b (Results 1 – 1 of 1) sorted by relevance
40 const auto& input_array_b = model->GetArray(input_rhs); in UnrollBatchMatMul3D() local78 std::vector<int> slice_size_b = input_array_b.shape().dims(); in UnrollBatchMatMul3D()91 slice_b_op_output.data_type = input_array_b.data_type; in UnrollBatchMatMul3D()99 {-1, input_array_b.shape().dims(dims_count - 1)})}; in UnrollBatchMatMul3D()104 slice_b_reshape_op_output.data_type = input_array_b.data_type; in UnrollBatchMatMul3D()247 const auto& input_array_b = model->GetArray(input_rhs); in Run() local251 CHECK_EQ(input_array_a.shape().dims(i), input_array_b.shape().dims(i)) in Run()255 input_array_b.shape().dims(dims - 2)) in Run()258 << absl::StrJoin(input_array_b.shape().dims(), ", ") << "]"; in Run()