Searched refs:with_flat_values (Results 1 – 14 of 14) sorted by relevance
73 return input.with_flat_values(string_bytes_split(input.flat_values))141 return input_tensor.with_flat_values(426 input = input.with_flat_values(456 codepoints = input.with_flat_values(codepoints)462 offsets = input.with_flat_values(offsets)513 return input.with_flat_values(889 str_t = rt.with_flat_values("'" + escaped + "'")891 str_t = rt.with_flat_values(string_ops.as_string(rt.flat_values))
149 return ragged_elts[0].with_flat_values(158 return x.with_flat_values(mapped_values)229 return x.with_flat_values(mapped_values)231 return y.with_flat_values(mapped_values)467 return x.with_flat_values(nn_ops.dropout(x.flat_values, keep_prob=keep_prob,476 return x.with_flat_values(nn_ops.dropout_v2(x.flat_values, rate=rate,
315 return total.with_flat_values(total.flat_values / count.flat_values)331 return total.with_flat_values(total.flat_values /
209 rt_int = rt.with_flat_values(tensor_int)212 rt_wrapped_int = rt.with_flat_values(WrappedTensor(tensor_int))
675 actual_values_grad = rt_value.with_flat_values(actual_flat_values_grad)711 return arg.with_flat_values(
560 rt_input = rt_input.with_flat_values(589 rt_input = rt_input.with_flat_values(
309 params_grad = params.with_flat_values(params_flat_values_grad)
409 result = indices.with_flat_values(gather_nd(params, indices.flat_values))
537 return indices.with_flat_values(
1336 def with_flat_values(self, new_values): member in RaggedTensor1353 return self.with_values(self.values.with_flat_values(new_values))
823 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"