Home
last modified time | relevance | path

Searched refs:serialized_sparse (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Ddeserialize_sparse_string_op.cc50 const Tensor& serialized_sparse = context->input(0); in Compute() local
51 const int ndims = serialized_sparse.shape().dims(); in Compute()
56 serialized_sparse.shape().DebugString())); in Compute()
58 OP_REQUIRES(context, serialized_sparse.shape().dim_size(ndims - 1) == 3, in Compute()
61 serialized_sparse.shape().DebugString())); in Compute()
65 num_sparse_tensors *= serialized_sparse.shape().dim_size(i); in Compute()
73 serialized_sparse.shape().DebugString())); in Compute()
78 const auto& serialized_sparse_t = serialized_sparse.vec<string>(); in Compute()
101 serialized_sparse.flat_inner_dims<string, 2>(); in Compute()
201 target_shape.vec<int64>()(i) = serialized_sparse.shape().dim_size(i); in Compute()
Dserialize_sparse_op.cc77 Tensor serialized_sparse; in Compute() local
78 OP_REQUIRES_OK(context, Initialize(&serialized_sparse)); in Compute()
80 auto serialized_sparse_t = serialized_sparse.vec<T>(); in Compute()
85 context->set_output(0, serialized_sparse); in Compute()
200 Tensor serialized_sparse; in Compute() local
201 OP_REQUIRES_OK(context, this->Initialize(N, &serialized_sparse)); in Compute()
202 auto serialized_sparse_t = serialized_sparse.matrix<U>(); in Compute()
259 context->set_output(0, serialized_sparse); in Compute()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dparse_example_dataset_op.cc316 Tensor& serialized_sparse = (*output)[output_index]; in MapFunc() local
317 auto serialized_sparse_t = serialized_sparse.vec<Variant>(); in MapFunc()
321 DCHECK(serialized_sparse.dtype() == in MapFunc()
326 << ", got " << DataTypeString(serialized_sparse.dtype()) in MapFunc()
329 serialized_sparse.shape())) in MapFunc()
333 << ", got " << serialized_sparse.shape().DebugString() in MapFunc()
Ddense_to_sparse_batch_dataset_op.cc271 Tensor serialized_sparse(DT_VARIANT, TensorShape({3})); in GetNextInternal() local
272 auto serialized_sparse_t = serialized_sparse.vec<Variant>(); in GetNextInternal()
276 out_tensors->push_back(std::move(serialized_sparse)); in GetNextInternal()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_DeserializeSparse.pbtxt4 name: "serialized_sparse"
18 The input `serialized_sparse` must have the shape `[?, ?, ..., ?, 3]` where
Dapi_def_DeserializeManySparse.pbtxt4 name: "serialized_sparse"
18 The input `serialized_sparse` must be a string matrix of shape `[N x 3]` where
Dapi_def_SerializeManySparse.pbtxt33 `SparseTensor` objects going into each row of `serialized_sparse` will have
/external/tensorflow/tensorflow/core/ops/
Dsparse_ops.cc178 ShapeHandle serialized_sparse; in __anona7ffb82a0802() local
179 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &serialized_sparse)); in __anona7ffb82a0802()
182 c->WithValue(c->Dim(serialized_sparse, 1), 3, &unused)); in __anona7ffb82a0802()
Dops.pbtxt9621 name: "serialized_sparse"
9644 name: "serialized_sparse"
34164 name: "serialized_sparse"
34200 name: "serialized_sparse"
/external/tensorflow/tensorflow/python/ops/
Dsparse_ops.py1989 def deserialize_sparse(serialized_sparse, dtype, rank=None, name=None): argument
2046 gen_sparse_ops.deserialize_sparse(serialized_sparse, dtype, name=name))
2059 def deserialize_many_sparse(serialized_sparse, dtype, rank=None, name=None): argument
2119 serialized_sparse, dtype, name=name))
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.io.pbtxt81 …argspec: "args=[\'serialized_sparse\', \'dtype\', \'rank\', \'name\'], varargs=None, keywords=None…
Dtensorflow.raw_ops.pbtxt885 …argspec: "args=[\'serialized_sparse\', \'dtype\', \'name\'], varargs=None, keywords=None, defaults…
889 …argspec: "args=[\'serialized_sparse\', \'dtype\', \'name\'], varargs=None, keywords=None, defaults…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.io.pbtxt97 …argspec: "args=[\'serialized_sparse\', \'dtype\', \'rank\', \'name\'], varargs=None, keywords=None…
Dtensorflow.pbtxt1081 …argspec: "args=[\'serialized_sparse\', \'dtype\', \'rank\', \'name\'], varargs=None, keywords=None…
Dtensorflow.raw_ops.pbtxt885 …argspec: "args=[\'serialized_sparse\', \'dtype\', \'name\'], varargs=None, keywords=None, defaults…
889 …argspec: "args=[\'serialized_sparse\', \'dtype\', \'name\'], varargs=None, keywords=None, defaults…
/external/tensorflow/tensorflow/go/op/
Dwrappers.go13517 func DeserializeManySparse(scope *Scope, serialized_sparse tf.Output, dtype tf.DataType) (sparse_in…
13525 serialized_sparse,
13581 func DeserializeSparse(scope *Scope, serialized_sparse tf.Output, dtype tf.DataType) (sparse_indice…
13589 serialized_sparse,
13779 …utput, sparse_shape tf.Output, optional ...SerializeManySparseAttr) (serialized_sparse tf.Output) {
37134 …tf.Output, sparse_shape tf.Output, optional ...SerializeSparseAttr) (serialized_sparse tf.Output) {
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v2.pbtxt20479 name: "serialized_sparse"
20502 name: "serialized_sparse"
20525 name: "serialized_sparse"
68837 name: "serialized_sparse"
68860 name: "serialized_sparse"
68896 name: "serialized_sparse"
68919 name: "serialized_sparse"
Dops_history.v1.pbtxt20822 name: "serialized_sparse"
20845 name: "serialized_sparse"
20868 name: "serialized_sparse"
69464 name: "serialized_sparse"
69487 name: "serialized_sparse"
69523 name: "serialized_sparse"
69546 name: "serialized_sparse"
Dops_history.v0.pbtxt10213 name: "serialized_sparse"
30519 name: "serialized_sparse"
30542 name: "serialized_sparse"