Searched refs:rt_nested_splits (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ragged_tensor_to_sparse_kernel.cc | 40 std::vector<TTypes<int64>::ConstFlat> rt_nested_splits; in Compute() local
41 rt_nested_splits.reserve(rt_nested_splits_len); in Compute()
43 rt_nested_splits.push_back(rt_nested_splits_in[i].flat<int64>()); in Compute()
49 ValidateInputs(rt_nested_splits, rt_dense_values_in)); in Compute()
62 (rt_nested_splits.back()(rt_nested_splits.back().size() - 1) * in Compute()
80 int max_final_pos = rt_nested_splits.back().size() - 1; in Compute()
85 while (IsCompleted(pos, dim, rt_nested_splits)) { in Compute()
92 int start = dim > 0 ? rt_nested_splits[dim - 1](pos[dim - 1]) : 0; in Compute()
97 const auto& final_splits = rt_nested_splits[rt_nested_splits_len - 1]; in Compute()
137 const auto& splits = rt_nested_splits[dim]; in Compute()
[all …]
|
| D | ragged_range_op.cc | 74 auto rt_nested_splits = rt_nested_splits_out->flat<int64>(); in Compute() local
75 rt_nested_splits(0) = 0; in Compute()
81 rt_nested_splits(row + 1) = in Compute()
82 rt_nested_splits(row) + RangeSize(start, limit, delta); in Compute()
84 int64 nvals = rt_nested_splits(nrows); in Compute()
93 int64 row_size = rt_nested_splits(row + 1) - rt_nested_splits(row); in Compute()
|
| D | ragged_tensor_to_sparse_kernel_test.cc | 39 const std::vector<std::vector<int64>>& rt_nested_splits, in BuildRaggedTensorToSparseGraph() argument
43 int64 num_splits = rt_nested_splits.size(); in BuildRaggedTensorToSparseGraph()
51 for (const auto& splits : rt_nested_splits) { in BuildRaggedTensorToSparseGraph()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_RaggedRange.pbtxt | 17 name: "rt_nested_splits"
30 `rt_nested_splits`, such that
34 >>> (rt_nested_splits, rt_dense_values) = gen_ragged_ops.ragged_range(
36 >>> result = ragged.from_nested_row_splits(rt_dense_values, rt_nested_splits)
|
| D | api_def_RaggedTensorToSparse.pbtxt | 5 name: "rt_nested_splits"
29 The ragged rank of the input RaggedTensor. `rt_nested_splits` should contain
38 input=ragged.from_nested_row_splits(rt_dense_values, rt_nested_splits)
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_tensor_value.py | 72 rt_nested_splits = [self.row_splits]
75 rt_nested_splits.append(rt_values.row_splits)
77 return tuple(rt_nested_splits)
|
| D | ragged_tensor.py | 769 rt_nested_splits = [self.row_splits]
772 rt_nested_splits.append(rt_values.row_splits)
774 return tuple(rt_nested_splits)
|
| D | ragged_math_ops.py | 104 result.rt_nested_splits)
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 12733 func RaggedTensorToSparse(scope *Scope, rt_nested_splits []tf.Output, rt_dense_values tf.Output) (s…
12740 tf.OutputList(rt_nested_splits), rt_dense_values,
35317 …cope *Scope, starts tf.Output, limits tf.Output, deltas tf.Output) (rt_nested_splits tf.Output, rt…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 2601 …argspec: "args=[\'rt_nested_splits\', \'rt_dense_values\', \'name\'], varargs=None, keywords=None,…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 2601 …argspec: "args=[\'rt_nested_splits\', \'rt_dense_values\', \'name\'], varargs=None, keywords=None,…
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 26533 name: "rt_nested_splits"
26560 name: "rt_nested_splits"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 51253 name: "rt_nested_splits"
51280 name: "rt_nested_splits"
|
| D | ops_history.v1.pbtxt | 51836 name: "rt_nested_splits"
51863 name: "rt_nested_splits"
|