Searched refs:with_values (Results 1 – 25 of 26) sorted by relevance
12
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_gather_ops.py | 142 return indices.with_values(_gather(params, indices.values, 0, 0))
214 return params.with_values(
224 adjusted_indices = params.with_values(
232 params.with_values(
272 return params.with_values(_gather(params.values, indices, axis - 1, 0))
317 return t.with_values(_increase_rank_to(t, rank - 1))
|
| D | ragged_getitem.py | 278 return rt_input.with_values(
318 return inner_rt.with_values(
458 return value_indices.with_values(gathered_values)
|
| D | ragged_tensor.py | 1290 return self.with_values(self.values.row_lengths(axis - 1))
1293 return self.with_values(
1382 def with_values(self, new_values): member in RaggedTensor
1407 return self.with_row_splits_dtype(dtypes.int64).with_values(new_values)
1428 return self.with_values(self.values.with_flat_values(new_values))
1431 return self.with_values(new_values)
2903 return value.with_values(
|
| D | ragged_dispatch.py | 32 return x.with_values(op(x.values))
|
| D | ragged_string_ops.py | 145 return input_tensor.with_values(
792 output = data.with_values(
|
| D | ragged_concat_ops.py | 309 rt_input = rt_input.with_values(
|
| D | ragged_math_ops.py | 603 return rt_input.with_values(
892 return a.with_values(flat_result)
1025 return a.with_values(array_ops.squeeze(flat_result, axis=1))
|
| D | ragged_tensor_supported_values_test.py | 167 rt_int = rt.with_values(tensor_int)
170 rt_wrapped_int = rt.with_values(WrappedTensor(tensor_int))
|
| D | ragged_array_ops.py | 457 return input.with_values(expand_dims(input.values, axis - 1))
459 return input.with_values(array_ops.expand_dims(input.values, axis - 1))
|
| D | ragged_tensor_test.py | 859 rt1_plus_10 = rt1.with_values(rt1.values + 10)
861 rt1_expanded = rt1.with_values(array_ops.expand_dims(rt1.values, axis=1))
1639 y = y.with_values(array_ops.stop_gradient(y.values))
1650 y = y.with_values(array_ops.stop_gradient(y.values))
|
| D | ragged_tensor_shape.py | 611 return rt_input.with_values(
|
| /external/tensorflow/tensorflow/python/kernel_tests/summary_ops/ |
| D | summary_v1_tensor_op_test.py | 131 with_values = summary_lib.tensor_summary(
137 descr = get_description(with_values)
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | composite_tensor_gradient.py | 105 return value.with_values(component_grads)
|
| D | sparse_tensor_test.py | 101 new_tensor = source.with_values([5.0, 1.0])
108 source.with_values([[5.0, 1.0]])
|
| D | sparse_tensor.py | 177 def with_values(self, new_values): member in SparseTensor
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-sparse-tensor.pbtxt | 56 name: "with_values"
|
| D | tensorflow.sparse.-sparse-tensor.pbtxt | 56 name: "with_values"
|
| D | tensorflow.-ragged-tensor.pbtxt | 156 name: "with_values"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.-sparse-tensor.pbtxt | 56 name: "with_values"
|
| D | tensorflow.sparse.-sparse-tensor.pbtxt | 56 name: "with_values"
|
| D | tensorflow.-ragged-tensor.pbtxt | 156 name: "with_values"
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/python/ |
| D | sentencepiece_tokenizer.py | 108 return input_tensor.with_values(tokens)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | parsing_config.py | 859 return partition_t.with_values(
865 return partition_t.with_values(
869 return partition_t.with_values(
|
| D | lookup_ops.py | 263 return keys.with_values(values)
1177 return keys.with_values(ids)
1395 return keys.with_values(ids)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | composite_tensor_ops_test.py | 100 return x3.with_values(x3.values * math_ops.range(6.0))
|
12