| /external/tensorflow/tensorflow/core/ops/ |
| D | ragged_conversion_ops.cc | 166 ShapeHandle rt_dense_values = c->input(num_splits); in RaggedTensorToSparseShapeFn() local
167 TF_RETURN_IF_ERROR(c->WithRankAtLeast(rt_dense_values, 1, &rt_dense_values)); in RaggedTensorToSparseShapeFn()
176 c->RankKnown(rt_dense_values) in RaggedTensorToSparseShapeFn()
177 ? c->MakeDim(c->Rank(rt_dense_values) + num_splits) in RaggedTensorToSparseShapeFn()
179 DimensionHandle num_values = c->NumElements(rt_dense_values); in RaggedTensorToSparseShapeFn()
193 shape_inference::ShapeHandle rt_dense_values = c->input(num_splits); in RaggedTensorToVariantShapeFn() local
194 TF_RETURN_IF_ERROR(c->WithRankAtLeast(rt_dense_values, 1, &rt_dense_values)); in RaggedTensorToVariantShapeFn()
|
| D | ops.pbtxt | 34940 name: "rt_dense_values"
35025 name: "rt_dense_values"
35130 name: "rt_dense_values"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_RaggedRange.pbtxt | 21 name: "rt_dense_values"
29 Returns a `RaggedTensor` `result` composed from `rt_dense_values` and
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 | 9 name: "rt_dense_values"
38 input=ragged.from_nested_row_splits(rt_dense_values, rt_nested_splits)
|
| D | api_def_RaggedTensorToVariant.pbtxt | 12 name: "rt_dense_values"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | RaggedTensorToVariant.pbtxt | 9 name: "rt_dense_values"
49 name: "rt_dense_values"
|
| D | RaggedTensorToSparse.pbtxt | 9 name: "rt_dense_values"
43 name: "rt_dense_values"
|
| D | RaggedRange.pbtxt | 20 name: "rt_dense_values"
59 name: "rt_dense_values"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | RaggedTensorToSparse.pbtxt | 9 name: "rt_dense_values"
43 name: "rt_dense_values"
|
| D | RaggedRange.pbtxt | 20 name: "rt_dense_values"
59 name: "rt_dense_values"
|
| D | RaggedTensorToVariant.pbtxt | 9 name: "rt_dense_values"
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ragged_range_op.cc | 90 auto rt_dense_values = rt_dense_values_out->flat<T>(); in Compute() local
97 rt_dense_values(value_index++) = T(value); in Compute()
|
| D | ragged_tensor_to_sparse_kernel_test.cc | 42 const std::vector<T>& rt_dense_values) { in BuildRaggedTensorToSparseGraph() argument
56 AddInputFromArray<T>(rt_dense_values_shape, rt_dense_values); in BuildRaggedTensorToSparseGraph()
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/ |
| D | ragged_range_tflite.cc | 119 T* rt_dense_values = GetTensorData<T>(&rt_dense_values_out); in EvalT() local
127 rt_dense_values[value_index++] = value; in EvalT()
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_util.py | 113 return array_ops.gather(params, offsets.rt_dense_values)
|
| D | ragged_conversion_ops.py | 167 rt_dense_values=grads[-1],
|
| D | ragged_math_ops.py | 112 result.rt_dense_values, result.rt_nested_splits, validate=False)
|
| /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 …s tf.Output, optional ...RaggedRangeAttr) (rt_nested_splits tf.Output, rt_dense_values tf.Output) {
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 | 10076 Returns a `RaggedTensor` `result` composed from `rt_dense_values` and
10081 (rt_nested_splits, rt_dense_values) = ragged_range(
10083 result = tf.ragged.from_row_splits(rt_dense_values, rt_nested_splits)
10101 …t64, TF_Int32, TF_Int64]>, [{The `flat_values` for the returned `RaggedTensor`.}]>:$rt_dense_values
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 50594 name: "rt_dense_values"
50622 name: "rt_dense_values"
|