Searched refs:with_flat_values (Results 1 – 15 of 15) sorted by relevance
72 return input.with_flat_values(string_bytes_split(input.flat_values))140 return input_tensor.with_flat_values(425 input = input.with_flat_values(455 codepoints = input.with_flat_values(codepoints)461 offsets = input.with_flat_values(offsets)512 return input.with_flat_values(900 str_t = rt.with_flat_values("'" + escaped + "'")902 str_t = rt.with_flat_values(string_ops.as_string(rt.flat_values))
76 return x.with_flat_values(mapped_values)78 return y.with_flat_values(mapped_values)
337 return total.with_flat_values(total.flat_values / count.flat_values)357 return total.with_flat_values(total.flat_values /1102 return x.with_flat_values(1118 return x.with_flat_values(1134 return x.with_flat_values(1234 return x.with_flat_values(
559 rt_input = rt_input.with_flat_values(588 rt_input = rt_input.with_flat_values(
181 rt_int = rt.with_flat_values(tensor_int)184 rt_wrapped_int = rt.with_flat_values(WrappedTensor(tensor_int))
347 params_grad = params.with_flat_values(params_flat_values_grad)
565 return indices.with_flat_values(1192 return input.with_flat_values(array_ops.bitcast(input.flat_values, type))
421 result = indices.with_flat_values(gather_nd(params, indices.flat_values))
715 return arg.with_flat_values(
1411 def with_flat_values(self, new_values): member in RaggedTensor1428 return self.with_values(self.values.with_flat_values(new_values))
1770 return x.with_flat_values(mapped_values)1772 return y.with_flat_values(mapped_values)
860 rt2_times_10 = rt2.with_flat_values(rt2.flat_values * 10)
60 return input_tensor.with_flat_values(tokens)104 return input_tensor.with_flat_values(tokens)
148 name: "with_flat_values"