Searched defs:SparseTensor (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/examples/custom_ops_doc/multiplex_3/ |
| D | multiplex_3_op.py | 41 def multiplex_sparse(cond: tf.SparseTensor,
42 a: tf.SparseTensor,
43 b: tf.SparseTensor,
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | sparse_tensor.h | 59 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
|
| D | README.md | 1 SparseTensor chapter
|
| D | sparse_tensor.cc | 98 SparseTensor::SparseTensor(Tensor ix, Tensor vals, const VarDimArray shape, in SparseTensor() function in tensorflow::sparse::SparseTensor
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_embedding_for_serving.py | 323 inp: sparse_tensor.SparseTensor,
324 weight: Optional[sparse_tensor.SparseTensor], table: tf_variables.Variable,
|
| D | tpu_embedding_v1.py | 307 self, inp: sparse_tensor.SparseTensor,
308 weight: Optional[sparse_tensor.SparseTensor],
|
| /external/pytorch/aten/src/ATen/native/ |
| D | SparseTensorUtils.h | 17 using SparseTensor = Tensor; variable
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | sparse_tensor.py | 43 class SparseTensor(internal.NativeObject, composite_tensor.CompositeTensor): class
|
| /external/pytorch/torch/testing/_internal/ |
| D | common_subclass.py | 100 class SparseTensor(WrapperTensor): class
|
| /external/pytorch/test/cpp/jit/ |
| D | test_misc.cpp | 2393 TEST(FutureTest, SparseTensor) { in TEST() argument
|