Searched defs:flat_values (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/ |
| H A D | ragged_tensor.py | 662 flat_values, argument
725 flat_values, argument
767 flat_values, argument
808 flat_values, argument
1048 def flat_values(self): member in RaggedTensor
|
| H A D | dynamic_ragged_shape.py | 1060 def _validate_flat_values_dynamically(self, flat_values): argument
1070 def _validate_flat_values(self, flat_values): argument
1091 def _add_row_partitions(self, flat_values, validate=False): argument
|
| H A D | ragged_tensor_value.py | 65 def flat_values(self): member in RaggedTensorValue
|
| H A D | ragged_tensor_supported_values_test.py | 423 def testToFromComponents(self, rt_spec, flat_values, nested_row_splits): argument
|
| H A D | ragged_math_ops.py | 1179 def _cumsum_flat_values_at_ragged_rank(last_rp, flat_values, exclusive=False, argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/utils/ |
| H A D | tensor_flatten.cpp | 98 auto flat_values = utils::flatten_dense_tensors(fmap(tensors, &get_values)); in flatten_sparse_tensors() local
104 const at::Tensor& flat_values, in unflatten_sparse_tensors()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/util/ |
| H A D | nest.cc | 97 auto flat_values = make_safe(swig::Flatten(value, false)); in FlattenDictItems() local
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A D | eval_frame.py | 1040 def check_user_input_output(flat_values, error_type): argument
|