Home
last modified time | relevance | path

Searched refs:sparse_to_dense (Results 1 – 25 of 30) sorted by relevance

12

/external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/
Dsparse_to_dense_op_py_test.py34 tf_ans = sparse_ops.sparse_to_dense([1, 3], [5], 1, 0)
39 tf_ans = sparse_ops.sparse_to_dense([1, 3], [5], 1.0, 0.0)
53 tf_ans = sparse_ops.sparse_to_dense(indices, [5], values, 0.0)
58 tf_ans = sparse_ops.sparse_to_dense([1, 3], [5], "a", "b")
63 tf_ans = sparse_ops.sparse_to_dense([1, 3], [5], [1, 2], -1)
68 tf_ans = sparse_ops.sparse_to_dense([1, 3], [5], 1, -1)
73 tf_ans = sparse_ops.sparse_to_dense([[1, 3], [2, 0]], [3, 4], 1, -1)
80 x = sparse_ops.sparse_to_dense(2, [4], 7)
84 tf_ans = sparse_ops.sparse_to_dense([[1, 3, 0], [2, 0, 1]], [3, 4, 2], 1,
94 sparse_ops.sparse_to_dense([1, 3], [[5], [3]], 1, -1)
[all …]
Dsparse_xent_op_test.py47 target = sparse_ops.sparse_to_dense(labels,
Dsparse_ops_test.py892 res_densified = sparse_ops.sparse_to_dense(
/external/tensorflow/tensorflow/compiler/tests/
Dsparse_to_dense_op_test.py34 return sparse_ops.sparse_to_dense(
79 x = sparse_ops.sparse_to_dense(2, [4], 7).eval()
/external/tensorflow/tensorflow/python/ops/
Dsparse_ops_test.py49 d = sparse_ops.sparse_to_dense(s.indices, s.dense_shape, s.values)
62 dense = sparse_ops.sparse_to_dense(sparse.indices, sparse.dense_shape,
103 d = sparse_ops.sparse_to_dense(s.indices, s.dense_shape, s.values)
155 dense_of_sparse = sparse_ops.sparse_to_dense(sp.indices, sp.shape,
Dsparse_ops.py1174 def sparse_to_dense(sparse_indices, function
1220 return gen_sparse_ops.sparse_to_dense(
1711 return gen_sparse_ops.sparse_to_dense(
Dnn_impl.py1984 sampled_logits += gen_sparse_ops.sparse_to_dense(
/external/tensorflow/tensorflow/lite/kernels/
Dsparse_to_dense.cc29 namespace sparse_to_dense { namespace
287 static TfLiteRegistration r = {nullptr, nullptr, sparse_to_dense::Prepare, in Register_SPARSE_TO_DENSE()
288 sparse_to_dense::Eval}; in Register_SPARSE_TO_DENSE()
DAndroid.bp155 "sparse_to_dense.cc",
/external/tensorflow/tensorflow/examples/adding_an_op/
Dzero_out_grad_2.py39 to_zero_grad = sparse_ops.sparse_to_dense([index], shape, first_grad, 0)
/external/tensorflow/tensorflow/lite/testing/op_tests/
Dsparse_to_dense.py66 out = tf.sparse_to_dense(
/external/pytorch/torch/testing/_internal/
Dcommon_subclass.py113 def sparse_to_dense(self): member in SparseTensor
135 return e.sparse_to_dense() if isinstance(e, SparseTensor) else e
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dinterleave_test.py244 sparse_ops.sparse_to_dense(x.indices, x.dense_shape, x.values))
439 sparse_ops.sparse_to_dense(x.indices, x.dense_shape, x.values))
499 sparse_ops.sparse_to_dense(x.indices, x.dense_shape, x.values))
Dflat_map_test.py124 sparse_ops.sparse_to_dense(x.indices, x.dense_shape, x.values))
288 sparse_ops.sparse_to_dense(x.indices, x.dense_shape, x.values))
/external/tensorflow/tensorflow/python/kernel_tests/array_ops/
Dscalar_test.py134 self.check(sparse_ops.sparse_to_dense, (1, 4, 7),
/external/tensorflow/tensorflow/lite/experimental/mlir/testing/
Dbuild_def.bzl79 "sparse_to_dense",
/external/tensorflow/tensorflow/lite/testing/
Dbuild_def.bzl152 "sparse_to_dense",
206 "sparse_to_dense",
Dgenerate_examples_lib.py163 from tensorflow.lite.testing.op_tests.sparse_to_dense import make_sparse_to_dense_tests
/external/tensorflow/tensorflow/lite/
Dtflite_static.bp151 "kernels/sparse_to_dense.cc",
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dparallel_interleave_test.py571 sparse_ops.sparse_to_dense(x.indices, x.dense_shape, x.values))
818 sparse_ops.sparse_to_dense(x.indices, x.dense_shape, x.values))
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dlegalize-tf-with-tf2xla.mlir184 // CHECK-LABEL: @sparse_to_dense
186 func.func @sparse_to_dense(%arg0: tensor<3x2xi32>, %arg1: tensor<3xf32>, %arg2: tensor<f32>) -> ten…
/external/tensorflow/tensorflow/python/kernel_tests/math_ops/
Dsets_test.py429 sparse_ops.sparse_to_dense(
447 sparse_ops.sparse_to_dense(
/external/pytorch/aten/src/ATen/native/
DTensorConversions.cpp624 Tensor sparse_to_dense(const Tensor& self, std::optional<ScalarType> dtype, std::optional<bool> mas… in sparse_to_dense() function
/external/tensorflow/tensorflow/compiler/mlir/tosa/tests/
Dtfl-to-tosa-pipeline.mlir1783 func.func @sparse_to_dense(%arg0 : tensor<?x2xi64>, %arg1 : tensor<?xi64>) -> (tensor<1x48xi64>) {
1786 …%2 = "tfl.sparse_to_dense"(%arg0, %0, %arg1, %1) : (tensor<?x2xi64>, tensor<2xi64>, tensor<?xi64>,…
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Dlegalize-tf.mlir1496 …// CHECK: "tfl.sparse_to_dense"(%arg0, %arg1, %arg2, %arg3) : (tensor<i32>, tensor<3xi32>, tensor<…
1503 …// CHECK: "tfl.sparse_to_dense"(%arg0, %arg1, %arg2, %arg3) : (tensor<3xi32>, tensor<3xi32>, tenso…
1510 …// CHECK: "tfl.sparse_to_dense"(%arg0, %arg1, %arg2, %arg3) : (tensor<3x2xi32>, tensor<3xi32>, ten…

12