Searched refs:DropFirstDim (Results 1 – 1 of 1) sorted by relevance
1058 Shape DropFirstDim(const Shape& shape) { in DropFirstDim() function1111 Shape inner_shape = DropFirstDim(outer_array.GetShape()); in SliceOutInnerArray()1162 dot_info.lhs_shape = DropFirstDim(lhs_array_reshaped.GetShape()); in EmitBatchDotOperation()1163 dot_info.rhs_shape = DropFirstDim(rhs_array_reshaped.GetShape()); in EmitBatchDotOperation()1164 dot_info.result_shape = DropFirstDim(target_array_reshaped.GetShape()); in EmitBatchDotOperation()