Home
last modified time | relevance | path

Searched refs:sparse_tensor_value (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dhandle_sparsetensors.cc201 mlir::Value sparse_tensor_value = in runOnOperation() local
212 sparse_tensor_value.getType() in runOnOperation()
218 front_op->getLoc(), sparse_tensor_value.getType(), in runOnOperation()
225 sparse_tensor_value.replaceAllUsesWith(sparse_to_dense_op); in runOnOperation()
226 if (!sparse_tensor_value.use_empty()) return signalPassFailure(); in runOnOperation()
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dsparse_tensor.py103 def from_value(cls, sparse_tensor_value): argument
104 if not is_sparse(sparse_tensor_value):
108 indices=sparse_tensor_value.indices,
109 values=sparse_tensor_value.values,
110 dense_shape=sparse_tensor_value.dense_shape)
H A Dsparse_tensor_test.py123 sparse_tensor_value = sparse_tensor.SparseTensorValue(
125 st = sparse_tensor.SparseTensor.from_value(sparse_tensor_value)
127 sparse_tensor.convert_to_tensor_or_sparse_tensor(sparse_tensor_value))
131 self.assertAllEqual(sparse_tensor_value.indices, convertee.indices)
132 self.assertAllEqual(sparse_tensor_value.values, convertee.values)
134 sparse_tensor_value.dense_shape, convertee.dense_shape)
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/math_ops/
H A Dsets_test.py1241 expected_shape, sparse_tensor_value, dtype): argument
1242 self.assertAllEqual(expected_indices, sparse_tensor_value.indices)
1244 self.assertAllEqual(len(expected_values), len(sparse_tensor_value.values))
1249 expected_indices, expected_values, sparse_tensor_value.values):
1264 self.assertAllEqual(expected_shape, sparse_tensor_value.dense_shape)
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.-sparse-tensor.pbtxt49 argspec: "args=[\'cls\', \'sparse_tensor_value\'], varargs=None, keywords=None, defaults=None"
H A Dtensorflow.sparse.-sparse-tensor.pbtxt49 argspec: "args=[\'cls\', \'sparse_tensor_value\'], varargs=None, keywords=None, defaults=None"
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/
H A Dtensorflow.-sparse-tensor.pbtxt49 argspec: "args=[\'cls\', \'sparse_tensor_value\'], varargs=None, keywords=None, defaults=None"
H A Dtensorflow.sparse.-sparse-tensor.pbtxt49 argspec: "args=[\'cls\', \'sparse_tensor_value\'], varargs=None, keywords=None, defaults=None"