/external/tensorflow/tensorflow/core/kernels/ |
D | deserialize_sparse_string_op.cc | 50 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()
|
D | serialize_sparse_op.cc | 77 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/ |
D | parse_example_dataset_op.cc | 316 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()
|
D | dense_to_sparse_batch_dataset_op.cc | 271 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/ |
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/ops/ |
D | sparse_ops.cc | 178 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()
|
D | ops.pbtxt | 9621 name: "serialized_sparse" 9644 name: "serialized_sparse" 34164 name: "serialized_sparse" 34200 name: "serialized_sparse"
|
/external/tensorflow/tensorflow/python/ops/ |
D | sparse_ops.py | 1989 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/ |
D | tensorflow.io.pbtxt | 81 …argspec: "args=[\'serialized_sparse\', \'dtype\', \'rank\', \'name\'], varargs=None, keywords=None…
|
D | tensorflow.raw_ops.pbtxt | 885 …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/ |
D | tensorflow.io.pbtxt | 97 …argspec: "args=[\'serialized_sparse\', \'dtype\', \'rank\', \'name\'], varargs=None, keywords=None…
|
D | tensorflow.pbtxt | 1081 …argspec: "args=[\'serialized_sparse\', \'dtype\', \'rank\', \'name\'], varargs=None, keywords=None…
|
D | tensorflow.raw_ops.pbtxt | 885 …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/ |
D | wrappers.go | 13517 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/ |
D | ops_history.v2.pbtxt | 20479 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"
|
D | ops_history.v1.pbtxt | 20822 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"
|
D | ops_history.v0.pbtxt | 10213 name: "serialized_sparse" 30519 name: "serialized_sparse" 30542 name: "serialized_sparse"
|