Searched refs:power_matrix (Results 1 – 1 of 1) sorted by relevance
146 def power_sums_tensor(array_size, power_matrix, multiplier): argument161 power_matrix = ops.convert_to_tensor(power_matrix)163 array_ops.shape(power_matrix)[0], dtype=power_matrix.dtype)165 ops.convert_to_tensor(power_matrix).get_shape())167 lambda previous_power, _: math_ops.matmul(previous_power, power_matrix),