| /external/tensorflow/tensorflow/core/kernels/ |
| D | ragged_tensor_from_variant_op_test.cc | 42 const int input_ragged_rank, const int output_ragged_rank, in BuildDecodeRaggedTensorGraph() argument
50 .Attr("output_ragged_rank", output_ragged_rank) in BuildDecodeRaggedTensorGraph()
94 int output_ragged_rank = 2; in TEST_F() local
96 output_ragged_rank, TensorShape({}), in TEST_F()
124 int output_ragged_rank = 3; in TEST_F() local
126 output_ragged_rank, TensorShape({1}), in TEST_F()
173 int output_ragged_rank = 2; in TEST_F() local
175 input_ragged_rank, output_ragged_rank, TensorShape({2, 2}), in TEST_F()
232 int output_ragged_rank = 3; in TEST_F() local
234 input_ragged_rank, output_ragged_rank, TensorShape({2, 5}), in TEST_F()
[all …]
|
| D | ragged_tensor_from_variant_op.cc | 83 const int output_ragged_rank, RaggedTensorVariant* output_ragged) { in NestedStackRaggedTensors() argument
84 output_ragged->mutable_nested_splits()->reserve(output_ragged_rank); in NestedStackRaggedTensors()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_RaggedTensorFromVariant.pbtxt | 27 -1, this is inferred as `output_ragged_rank` - `rank(encoded_ragged)`
31 name: "output_ragged_rank"
34 `output_ragged_rank = rank(encoded_ragged) + input_ragged_rank`.
43 `output_ragged_rank`. It could also have an arbitrary rank, in which case each
46 `RaggedTensor` with ragged_rank `output_ragged_rank`. Each `variant` element in
50 inferred as `output_ragged_rank` - `rank(encoded_ragged)`. See
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | RaggedTensorFromVariant.pbtxt | 10 number_attr: "output_ragged_rank"
23 name: "output_ragged_rank"
52 number_attr: "output_ragged_rank"
65 name: "output_ragged_rank"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ragged_conversion_ops.cc | 226 int64 output_ragged_rank; in RaggedTensorFromVariantShapeFn() local
228 c->GetAttr<int64>("output_ragged_rank", &output_ragged_rank)); in RaggedTensorFromVariantShapeFn()
233 encoded_ragged, output_ragged_rank - input_ragged_rank, &unused)); in RaggedTensorFromVariantShapeFn()
235 for (int64 i = 0; i < output_ragged_rank; i++) { in RaggedTensorFromVariantShapeFn()
238 c->set_output(output_ragged_rank, c->UnknownShape()); in RaggedTensorFromVariantShapeFn()
|
| D | ops.pbtxt | 34982 number_attr: "output_ragged_rank"
34995 name: "output_ragged_rank"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | RaggedTensorFromVariant.pbtxt | 10 number_attr: "output_ragged_rank"
23 name: "output_ragged_rank"
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_tensor_test.py | 1377 et, dtype, output_ragged_rank=ragged_rank)
1390 output_ragged_rank=ragged_rank + 1)
1406 output_ragged_rank=ragged_rank + 1,
1418 batched[i], dtype=dtypes.int32, output_ragged_rank=0)
1443 output_ragged_rank=1,
1463 rt3 = RaggedTensor._from_variant(v, dtype=rt2.dtype, output_ragged_rank=1)
1476 rt3 = RaggedTensor._from_variant(v, dtype=rt2.dtype, output_ragged_rank=1)
1490 output_ragged_rank=0)
1510 return RaggedTensor._from_variant(v, dtype=x2.dtype, output_ragged_rank=0)
1523 rt3 = RaggedTensor._from_variant(v, dtype=x2.dtype, output_ragged_rank=3)
[all …]
|
| D | ragged_tensor.py | 1866 output_ragged_rank, argument
1916 output_ragged_rank != input_ragged_rank + variant.shape.ndims):
1921 (variant.shape.ndims, input_ragged_rank, output_ragged_rank))
1925 [variant, dtype, input_ragged_rank, output_ragged_rank]):
1927 variant, input_ragged_rank, output_ragged_rank, dtype,
2443 output_ragged_rank=self._ragged_rank)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 3225 …argspec: "args=[\'encoded_ragged\', \'input_ragged_rank\', \'output_ragged_rank\', \'Tvalues\', \'…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 3225 …argspec: "args=[\'encoded_ragged\', \'input_ragged_rank\', \'output_ragged_rank\', \'Tvalues\', \'…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 27881 …t(scope *Scope, encoded_ragged tf.Output, input_ragged_rank int64, output_ragged_rank int64, Tvalu…
27885 …ce{}{"input_ragged_rank": input_ragged_rank, "output_ragged_rank": output_ragged_rank, "Tvalues": …
|