Searched refs:transpose_dims (Results 1 – 3 of 3) sorted by relevance
42 transpose_dims = [dim_map[d] for d in data_format_dst]43 return np.transpose(x, transpose_dims)
243 std::vector<int64> transpose_dims(output_rank); in Einsum() local245 transpose_dims[output_dims[i]] = i; in Einsum()251 return Transpose(DotGeneral(x, y, dnums, &precision_proto), transpose_dims); in Einsum()
463 transpose_dims = array_ops.concat(465 values_transpose = array_ops.transpose(values, transpose_dims)