Home
last modified time | relevance | path

Searched defs:SparseTensor (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/examples/custom_ops_doc/multiplex_3/
Dmultiplex_3_op.py41 def multiplex_sparse(cond: tf.SparseTensor,
42 a: tf.SparseTensor,
43 b: tf.SparseTensor,
/external/tensorflow/tensorflow/core/util/sparse/
Dsparse_tensor.h59 SparseTensor() : dims_(0) {} in SparseTensor() function
62 SparseTensor(Tensor ix, Tensor vals, const TensorShape& shape) in SparseTensor() function
67 SparseTensor(Tensor ix, Tensor vals, const VarDimArray shape) in SparseTensor() function
72 SparseTensor(Tensor ix, Tensor vals, const TensorShape& shape, in SparseTensor() function
81 SparseTensor(const SparseTensor& other) in SparseTensor() function
84 SparseTensor(SparseTensor&& other) in SparseTensor() function
DREADME.md1 SparseTensor chapter
Dsparse_tensor.cc98 SparseTensor::SparseTensor(Tensor ix, Tensor vals, const VarDimArray shape, in SparseTensor() function in tensorflow::sparse::SparseTensor
/external/tensorflow/tensorflow/python/tpu/
Dtpu_embedding_for_serving.py323 inp: sparse_tensor.SparseTensor,
324 weight: Optional[sparse_tensor.SparseTensor], table: tf_variables.Variable,
Dtpu_embedding_v1.py307 self, inp: sparse_tensor.SparseTensor,
308 weight: Optional[sparse_tensor.SparseTensor],
/external/tensorflow/tensorflow/python/framework/
Dsparse_tensor.py43 class SparseTensor(internal.NativeObject, composite_tensor.CompositeTensor): class