Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/util/sparse/
H A 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
H A DREADME.md1 SparseTensor chapter
H A Dsparse_tensor.cc98 SparseTensor::SparseTensor(Tensor ix, Tensor vals, const VarDimArray shape, in SparseTensor() function in tensorflow::sparse::SparseTensor
/aosp_15_r20/external/tensorflow/tensorflow/examples/custom_ops_doc/multiplex_3/
H A Dmultiplex_3_op.py41 def multiplex_sparse(cond: tf.SparseTensor,
42 a: tf.SparseTensor,
43 b: tf.SparseTensor,
/aosp_15_r20/external/tensorflow/tensorflow/python/tpu/
H A Dtpu_embedding_for_serving.py323 inp: sparse_tensor.SparseTensor,
324 weight: Optional[sparse_tensor.SparseTensor], table: tf_variables.Variable,
H A Dtpu_embedding_v1.py307 self, inp: sparse_tensor.SparseTensor,
308 weight: Optional[sparse_tensor.SparseTensor],
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DSparseTensorUtils.h17 using SparseTensor = Tensor; variable
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dsparse_tensor.py43 class SparseTensor(internal.NativeObject, composite_tensor.CompositeTensor): class
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_subclass.py100 class SparseTensor(WrapperTensor): class
/aosp_15_r20/external/pytorch/test/cpp/jit/
H A Dtest_misc.cpp2393 TEST(FutureTest, SparseTensor) { in TEST() argument