Searched refs:DropFirstDim (Results 1 – 1 of 1) sorted by relevance
862 Shape DropFirstDim(const Shape& shape) { in DropFirstDim() function915 Shape inner_shape = DropFirstDim(outer_array.GetShape()); in SliceOutInnerArray()966 dot_info.lhs_shape = DropFirstDim(lhs_array_reshaped.GetShape()); in EmitBatchDotOperation()967 dot_info.rhs_shape = DropFirstDim(rhs_array_reshaped.GetShape()); in EmitBatchDotOperation()968 dot_info.result_shape = DropFirstDim(target_array_reshaped.GetShape()); in EmitBatchDotOperation()