Searched refs:DropFirstDim (Results 1 – 1 of 1) sorted by relevance
1255 Shape DropFirstDim(const Shape& shape) { in DropFirstDim() function1309 Shape inner_shape = DropFirstDim(outer_array.GetShape()); in SliceOutInnerArray()1351 dot_info.lhs_shape = DropFirstDim(lhs_array_reshaped.GetShape()); in PotentiallyImplementedAsEigenMatmul()1352 dot_info.rhs_shape = DropFirstDim(rhs_array_reshaped.GetShape()); in PotentiallyImplementedAsEigenMatmul()1353 dot_info.result_shape = DropFirstDim(target_array_reshaped.GetShape()); in PotentiallyImplementedAsEigenMatmul()1432 dot_info.lhs_shape = DropFirstDim(lhs_array_reshaped.GetShape()); in EmitBatchDotOperation()1433 dot_info.rhs_shape = DropFirstDim(rhs_array_reshaped.GetShape()); in EmitBatchDotOperation()1435 DropFirstDim(target_array_reshaped.GetShape()); in EmitBatchDotOperation()