Searched refs:with_flat_values (Results 1 – 9 of 9) sorted by relevance
79 return input_tensor.with_flat_values(361 input = input.with_flat_values(391 codepoints = input.with_flat_values(codepoints)396 offsets = input.with_flat_values(offsets)
147 return x.with_flat_values(mapped_values)215 return x.with_flat_values(mapped_values)217 return y.with_flat_values(mapped_values)
282 return total.with_flat_values(total.flat_values / count.flat_values)296 return total.with_flat_values(
201 result = indices.with_flat_values(gather_nd(params, indices.flat_values))
511 rt_input = rt_input.with_flat_values(534 rt_input = rt_input.with_flat_values(
1020 def with_flat_values(self, new_values): member in RaggedTensor1039 return self.with_values(self.values.with_flat_values(new_values))
1123 rt2_times_10 = rt2.with_flat_values(rt2.flat_values * 10)
123 name: "with_flat_values"