Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/kernels/
Ddeserialize_sparse_string_op.cc52 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_t>()(i) = serialized_sparse.shape().dim_size(i); in Compute()
Dserialize_sparse_op.cc81 Tensor serialized_sparse; in Compute() local
82 OP_REQUIRES_OK(context, Initialize(&serialized_sparse)); in Compute()
84 auto serialized_sparse_t = serialized_sparse.vec<T>(); in Compute()
89 context->set_output(0, serialized_sparse); in Compute()
153 Tensor* serialized_sparse) { in operator ()()
154 auto serialized_sparse_t = serialized_sparse->matrix<tstring>(); in operator ()()
257 Tensor* serialized_sparse) { in operator ()()
258 auto serialized_sparse_t = serialized_sparse->template matrix<Variant>(); in operator ()()
385 Tensor* serialized_sparse; in Compute() local
387 context->allocate_output(0, {N, 3}, &serialized_sparse)); in Compute()
[all …]
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DSerializeManySparse.pbtxt16 name: "serialized_sparse"
39 name: "serialized_sparse"
DDeserializeSparse.pbtxt4 name: "serialized_sparse"
27 name: "serialized_sparse"
DSerializeSparse.pbtxt16 name: "serialized_sparse"
39 name: "serialized_sparse"
DDeserializeManySparse.pbtxt4 name: "serialized_sparse"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DSerializeSparse.pbtxt16 name: "serialized_sparse"
39 name: "serialized_sparse"
DDeserializeSparse.pbtxt4 name: "serialized_sparse"
27 name: "serialized_sparse"
DSerializeManySparse.pbtxt16 name: "serialized_sparse"
39 name: "serialized_sparse"
DDeserializeManySparse.pbtxt4 name: "serialized_sparse"
/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/kernels/data/experimental/
Ddense_to_sparse_batch_dataset_op.cc279 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()
Dparse_example_dataset_op.cc682 Tensor& serialized_sparse = (*output)[output_index]; in ParseExample() local
683 auto serialized_sparse_t = serialized_sparse.vec<Variant>(); in ParseExample()
688 CheckOutputTensor(serialized_sparse, d, output_index)); in ParseExample()
/external/tensorflow/tensorflow/security/advisory/
Dtfsa-2021-181.md17 serialized_sparse=tf.data.experimental.to_variant(dataset),
/external/tensorflow/tensorflow/core/ops/
Dsparse_ops.cc182 ShapeHandle serialized_sparse; in __anonad4a10b70802() local
183 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &serialized_sparse)); in __anonad4a10b70802()
186 c->WithValue(c->Dim(serialized_sparse, 1), 3, &unused)); in __anonad4a10b70802()
/external/tensorflow/tensorflow/python/ops/
Dsparse_ops.py2272 def deserialize_sparse(serialized_sparse, dtype, rank=None, name=None): argument
2329 gen_sparse_ops.deserialize_sparse(serialized_sparse, dtype, name=name))
2343 def deserialize_many_sparse(serialized_sparse, dtype, rank=None, name=None): argument
2403 serialized_sparse, dtype, name=name))
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.io.pbtxt85 …argspec: "args=[\'serialized_sparse\', \'dtype\', \'rank\', \'name\'], varargs=None, keywords=None…
Dtensorflow.raw_ops.pbtxt1273 …argspec: "args=[\'serialized_sparse\', \'dtype\', \'name\'], varargs=None, keywords=None, defaults…
1277 …argspec: "args=[\'serialized_sparse\', \'dtype\', \'name\'], varargs=None, keywords=None, defaults…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.io.pbtxt105 …argspec: "args=[\'serialized_sparse\', \'dtype\', \'rank\', \'name\'], varargs=None, keywords=None…
Dtensorflow.pbtxt1125 …argspec: "args=[\'serialized_sparse\', \'dtype\', \'rank\', \'name\'], varargs=None, keywords=None…
Dtensorflow.raw_ops.pbtxt1273 …argspec: "args=[\'serialized_sparse\', \'dtype\', \'name\'], varargs=None, keywords=None, defaults…
1277 …argspec: "args=[\'serialized_sparse\', \'dtype\', \'name\'], varargs=None, keywords=None, defaults…
/external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/
Dsparse_serialization_ops_test.py472 serialized_sparse=mu_lock, dtype=dtypes.int32)
/external/tensorflow/tensorflow/go/op/
Dwrappers.go12663 func DeserializeManySparse(scope *Scope, serialized_sparse tf.Output, dtype tf.DataType) (sparse_in…
12671 serialized_sparse,
12730 func DeserializeSparse(scope *Scope, serialized_sparse tf.Output, dtype tf.DataType) (sparse_indice…
12738 serialized_sparse,
42689 …utput, sparse_shape tf.Output, optional ...SerializeManySparseAttr) (serialized_sparse tf.Output) {
42729 …tf.Output, sparse_shape tf.Output, optional ...SerializeSparseAttr) (serialized_sparse tf.Output) {

12