| /external/tensorflow/tensorflow/core/kernels/ |
| D | deserialize_sparse_string_op.cc | 52 const Tensor& serialized_sparse = context->input(0); in Compute() local
53 const int ndims = serialized_sparse.shape().dims(); in Compute()
58 serialized_sparse.shape().DebugString())); in Compute()
60 OP_REQUIRES(context, serialized_sparse.shape().dim_size(ndims - 1) == 3, in Compute()
63 serialized_sparse.shape().DebugString())); in Compute()
67 num_sparse_tensors *= serialized_sparse.shape().dim_size(i); in Compute()
75 serialized_sparse.shape().DebugString())); in Compute()
80 const auto& serialized_sparse_t = serialized_sparse.vec<tstring>(); in Compute()
103 serialized_sparse.flat_inner_dims<tstring, 2>(); in Compute()
203 target_shape.vec<int64>()(i) = serialized_sparse.shape().dim_size(i); in Compute()
|
| D | serialize_sparse_op.cc | 79 Tensor serialized_sparse; in Compute() local
80 OP_REQUIRES_OK(context, Initialize(&serialized_sparse)); in Compute()
82 auto serialized_sparse_t = serialized_sparse.vec<T>(); in Compute()
87 context->set_output(0, serialized_sparse); in Compute()
151 Tensor* serialized_sparse) { in operator ()()
152 auto serialized_sparse_t = serialized_sparse->matrix<tstring>(); in operator ()()
253 Tensor* serialized_sparse) { in operator ()()
254 auto serialized_sparse_t = serialized_sparse->template matrix<Variant>(); in operator ()()
378 Tensor* serialized_sparse; in Compute() local
380 context->allocate_output(0, {N, 3}, &serialized_sparse)); in Compute()
[all …]
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | SerializeManySparse.pbtxt | 16 name: "serialized_sparse"
39 name: "serialized_sparse"
|
| D | SerializeSparse.pbtxt | 16 name: "serialized_sparse"
39 name: "serialized_sparse"
|
| D | DeserializeSparse.pbtxt | 4 name: "serialized_sparse"
27 name: "serialized_sparse"
|
| D | DeserializeManySparse.pbtxt | 4 name: "serialized_sparse"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | DeserializeSparse.pbtxt | 4 name: "serialized_sparse"
27 name: "serialized_sparse"
|
| D | SerializeManySparse.pbtxt | 16 name: "serialized_sparse"
39 name: "serialized_sparse"
|
| D | SerializeSparse.pbtxt | 16 name: "serialized_sparse"
39 name: "serialized_sparse"
|
| D | DeserializeManySparse.pbtxt | 4 name: "serialized_sparse"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_DeserializeSparse.pbtxt | 4 name: "serialized_sparse"
18 The input `serialized_sparse` must have the shape `[?, ?, ..., ?, 3]` where
|
| D | api_def_DeserializeManySparse.pbtxt | 4 name: "serialized_sparse"
18 The input `serialized_sparse` must be a string matrix of shape `[N x 3]` where
|
| D | api_def_SerializeManySparse.pbtxt | 33 `SparseTensor` objects going into each row of `serialized_sparse` will have
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | dense_to_sparse_batch_dataset_op.cc | 279 Tensor serialized_sparse(DT_VARIANT, TensorShape({3})); in GetNextInternal() local
280 auto serialized_sparse_t = serialized_sparse.vec<Variant>(); in GetNextInternal()
284 out_tensors->push_back(std::move(serialized_sparse)); in GetNextInternal()
|
| D | parse_example_dataset_op.cc | 671 Tensor& serialized_sparse = (*output)[output_index]; in ParseExample() local
672 auto serialized_sparse_t = serialized_sparse.vec<Variant>(); in ParseExample()
677 CheckOutputTensor(serialized_sparse, d, output_index)); in ParseExample()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | sparse_ops.cc | 178 ShapeHandle serialized_sparse; in __anon36a8fc880802() local
179 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &serialized_sparse)); in __anon36a8fc880802()
182 c->WithValue(c->Dim(serialized_sparse, 1), 3, &unused)); in __anon36a8fc880802()
|
| D | ops.pbtxt | 12934 name: "serialized_sparse"
12957 name: "serialized_sparse"
45589 name: "serialized_sparse"
45625 name: "serialized_sparse"
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | sparse_ops.py | 2279 def deserialize_sparse(serialized_sparse, dtype, rank=None, name=None): argument
2336 gen_sparse_ops.deserialize_sparse(serialized_sparse, dtype, name=name))
2350 def deserialize_many_sparse(serialized_sparse, dtype, rank=None, name=None): argument
2410 serialized_sparse, dtype, name=name))
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.io.pbtxt | 85 …argspec: "args=[\'serialized_sparse\', \'dtype\', \'rank\', \'name\'], varargs=None, keywords=None…
|
| D | tensorflow.raw_ops.pbtxt | 1205 …argspec: "args=[\'serialized_sparse\', \'dtype\', \'name\'], varargs=None, keywords=None, defaults…
1209 …argspec: "args=[\'serialized_sparse\', \'dtype\', \'name\'], varargs=None, keywords=None, defaults…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.io.pbtxt | 105 …argspec: "args=[\'serialized_sparse\', \'dtype\', \'rank\', \'name\'], varargs=None, keywords=None…
|
| D | tensorflow.pbtxt | 1117 …argspec: "args=[\'serialized_sparse\', \'dtype\', \'rank\', \'name\'], varargs=None, keywords=None…
|
| D | tensorflow.raw_ops.pbtxt | 1205 …argspec: "args=[\'serialized_sparse\', \'dtype\', \'name\'], varargs=None, keywords=None, defaults…
1209 …argspec: "args=[\'serialized_sparse\', \'dtype\', \'name\'], varargs=None, keywords=None, defaults…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 32638 func DeserializeManySparse(scope *Scope, serialized_sparse tf.Output, dtype tf.DataType) (sparse_in…
32646 serialized_sparse,
47201 …tf.Output, sparse_shape tf.Output, optional ...SerializeSparseAttr) (serialized_sparse tf.Output) {
49431 func DeserializeSparse(scope *Scope, serialized_sparse tf.Output, dtype tf.DataType) (sparse_indice…
49439 serialized_sparse,
49725 …utput, sparse_shape tf.Output, optional ...SerializeManySparseAttr) (serialized_sparse tf.Output) {
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 20479 name: "serialized_sparse"
20502 name: "serialized_sparse"
20525 name: "serialized_sparse"
68174 name: "serialized_sparse"
68197 name: "serialized_sparse"
68233 name: "serialized_sparse"
68256 name: "serialized_sparse"
|