| /external/tensorflow/tensorflow/core/kernels/ |
| D | ragged_tensor_to_sparse_kernel.cc | 42 std::vector<ConstFlatSplits> rt_nested_splits; in Compute() local
43 rt_nested_splits.reserve(rt_nested_splits_len); in Compute()
45 rt_nested_splits.push_back(rt_nested_splits_in[i].flat<SPLITS_TYPE>()); in Compute()
51 ValidateInputs(rt_nested_splits, rt_dense_values_in)); in Compute()
64 (rt_nested_splits.back()(rt_nested_splits.back().size() - 1) * in Compute()
82 int max_final_pos = rt_nested_splits.back().size() - 1; in Compute()
87 while (IsCompleted(pos, dim, rt_nested_splits)) { in Compute()
94 int start = dim > 0 ? rt_nested_splits[dim - 1](pos[dim - 1]) : 0; in Compute()
99 const auto& final_splits = rt_nested_splits[rt_nested_splits_len - 1]; in Compute()
139 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<SPLITS_TYPE>(); 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 SPLITS_TYPE nvals = rt_nested_splits(nrows); in Compute()
93 SPLITS_TYPE row_size = rt_nested_splits(row + 1) - rt_nested_splits(row); in Compute()
|
| D | ragged_tensor_to_sparse_kernel_test.cc | 40 const std::vector<std::vector<int64>>& rt_nested_splits, in BuildRaggedTensorToSparseGraph() argument
44 int64 num_splits = rt_nested_splits.size(); in BuildRaggedTensorToSparseGraph()
52 for (const auto& splits : rt_nested_splits) { in BuildRaggedTensorToSparseGraph()
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/ |
| D | ragged_range_tflite.cc | 97 SPLITS_TYPE* rt_nested_splits = in EvalT() local
99 rt_nested_splits[0] = 0; in EvalT()
109 rt_nested_splits[row + 1] = in EvalT()
110 rt_nested_splits[row] + RangeSize<T, SPLITS_TYPE>(start, limit, delta); in EvalT()
112 const SPLITS_TYPE nvals = rt_nested_splits[nrows]; in EvalT()
123 rt_nested_splits[row + 1] - rt_nested_splits[row]; in EvalT()
|
| /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) = ragged_range(
36 result = tf.ragged.from_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)
|
| D | api_def_RaggedTensorToVariant.pbtxt | 5 name: "rt_nested_splits"
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_tensor_value.py | 76 rt_nested_splits = [self.row_splits]
79 rt_nested_splits.append(rt_values.row_splits)
81 return tuple(rt_nested_splits)
|
| D | ragged_conversion_ops.py | 166 rt_nested_splits=op.outputs[:-1],
|
| D | ragged_math_ops.py | 112 result.rt_dense_values, result.rt_nested_splits, validate=False)
|
| D | ragged_tensor.py | 1032 rt_nested_splits = [self.row_splits]
1035 rt_nested_splits.append(rt_values.row_splits)
1037 return tuple(rt_nested_splits)
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | RaggedTensorToVariant.pbtxt | 4 name: "rt_nested_splits"
44 name: "rt_nested_splits"
|
| D | RaggedTensorToSparse.pbtxt | 4 name: "rt_nested_splits"
38 name: "rt_nested_splits"
|
| D | RaggedRange.pbtxt | 16 name: "rt_nested_splits"
55 name: "rt_nested_splits"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | RaggedTensorToSparse.pbtxt | 4 name: "rt_nested_splits"
38 name: "rt_nested_splits"
|
| D | RaggedRange.pbtxt | 16 name: "rt_nested_splits"
55 name: "rt_nested_splits"
|
| D | RaggedTensorToVariant.pbtxt | 4 name: "rt_nested_splits"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 28304 func RaggedTensorToSparse(scope *Scope, rt_nested_splits []tf.Output, rt_dense_values tf.Output) (s…
28311 tf.OutputList(rt_nested_splits), rt_dense_values,
28461 …, limits tf.Output, deltas tf.Output, optional ...RaggedRangeAttr) (rt_nested_splits tf.Output, rt…
42222 func RaggedTensorToVariant(scope *Scope, rt_nested_splits []tf.Output, rt_dense_values tf.Output, b…
42230 tf.OutputList(rt_nested_splits), rt_dense_values,
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 3229 …argspec: "args=[\'rt_nested_splits\', \'rt_dense_values\', \'name\'], varargs=None, keywords=None,…
3237 …argspec: "args=[\'rt_nested_splits\', \'rt_dense_values\', \'batched_input\', \'name\'], varargs=N…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 3229 …argspec: "args=[\'rt_nested_splits\', \'rt_dense_values\', \'name\'], varargs=None, keywords=None,…
3237 …argspec: "args=[\'rt_nested_splits\', \'rt_dense_values\', \'batched_input\', \'name\'], varargs=N…
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 10077 `rt_nested_splits`, such that
10081 (rt_nested_splits, rt_dense_values) = ragged_range(
10083 result = tf.ragged.from_row_splits(rt_dense_values, rt_nested_splits)
10100 Res<TF_I32OrI64Tensor, [{The `row_splits` for the returned `RaggedTensor`.}]>:$rt_nested_splits,
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 34936 name: "rt_nested_splits"
35020 name: "rt_nested_splits"
35125 name: "rt_nested_splits"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 50590 name: "rt_nested_splits"
50617 name: "rt_nested_splits"
|