Home
last modified time | relevance | path

Searched refs:with_values (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_gather_ops.py144 return indices.with_values(_gather(params, indices.values, 0, 0))
216 return params.with_values(
226 adjusted_indices = params.with_values(
234 params.with_values(
274 return params.with_values(_gather(params.values, indices, axis - 1, 0))
319 return t.with_values(_increase_rank_to(t, rank - 1))
Dragged_getitem.py276 return rt_input.with_values(
316 return inner_rt.with_values(
456 return value_indices.with_values(gathered_values)
Dragged_tensor.py1218 return self.with_values(self.values.row_lengths(axis - 1))
1221 return self.with_values(
1307 def with_values(self, new_values): member in RaggedTensor
1332 return self.with_row_splits_dtype(dtypes.int64).with_values(new_values)
1353 return self.with_values(self.values.with_flat_values(new_values))
1356 return self.with_values(new_values)
2752 return value.with_values(
Dragged_concat_ops.py307 rt_input = rt_input.with_values(
Dragged_tensor_supported_values_test.py195 rt_int = rt.with_values(tensor_int)
198 rt_wrapped_int = rt.with_values(WrappedTensor(tensor_int))
Dragged_array_ops.py451 return input.with_values(expand_dims(input.values, axis - 1))
453 return input.with_values(array_ops.expand_dims(input.values, axis - 1))
Dragged_string_ops.py146 return input_tensor.with_values(
788 output = data.with_values(
Dragged_math_ops.py561 return rt_input.with_values(
Dragged_tensor_shape.py612 return rt_input.with_values(
Dragged_tensor_test.py822 rt1_plus_10 = rt1.with_values(rt1.values + 10)
824 rt1_expanded = rt1.with_values(array_ops.expand_dims(rt1.values, axis=1))
/external/tensorflow/tensorflow/python/kernel_tests/
Dsummary_v1_tensor_op_test.py136 with_values = summary_lib.tensor_summary(
142 descr = get_description(with_values)
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.-sparse-tensor.pbtxt56 name: "with_values"
Dtensorflow.sparse.-sparse-tensor.pbtxt56 name: "with_values"
Dtensorflow.-ragged-tensor.pbtxt156 name: "with_values"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.sparse.-sparse-tensor.pbtxt56 name: "with_values"
Dtensorflow.-sparse-tensor.pbtxt56 name: "with_values"
Dtensorflow.-ragged-tensor.pbtxt156 name: "with_values"
/external/tflite-support/tensorflow_lite_support/custom_ops/python/
Dsentencepiece_tokenizer.py108 return input_tensor.with_values(tokens)
/external/tensorflow/tensorflow/python/ops/
Dparsing_config.py853 return partition_t.with_values(
859 return partition_t.with_values(
863 return partition_t.with_values(
Dlookup_ops.py244 return keys.with_values(values)
1143 return keys.with_values(ids)
1336 return keys.with_values(ids)
/external/tensorflow/tensorflow/python/framework/
Dsparse_tensor_test.py103 new_tensor = source.with_values([5.0, 1.0])
110 source.with_values([[5.0, 1.0]])
Dsparse_tensor.py181 def with_values(self, new_values): member in SparseTensor
/external/tensorflow/tensorflow/python/tpu/
Dtpu_embedding_v2_test.py1089 weights.append(ragged.with_values(
/external/tensorflow/
DRELEASE.md498 …* Adds `tf.SparseTensor.with_values`. This returns a new SparseTensor with the same sparsity patte…
499 similar to the `with_values` function of `RaggedTensor`.