Searched refs:sparse_output (Results 1 – 8 of 8) sorted by relevance
| /external/pytorch/benchmarks/sparse/ |
| D | benchmark_semi_structured_sparsity.py | 74 sparse_output = model(input_tensor)
75 print(sparse_output.shape)
82 correct = torch.allclose(dense_output, sparse_output, rtol=1e-3, atol=1e-3)
95 "contiguous": sparse_output.is_contiguous(),
124 sparse_output = torch.mm(sA, B)
130 correct = torch.allclose(dense_output, sparse_output, rtol=1e-3, atol=1e-3)
143 "contiguous": sparse_output.is_contiguous(),
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | DenseToCSRSparseMatrix.pbtxt | 12 name: "sparse_output"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | DenseToCSRSparseMatrix.pbtxt | 12 name: "sparse_output"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_DenseToCSRSparseMatrix.pbtxt | 13 name: "sparse_output"
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | tensor_utils_test.cc | 1549 std::vector<float> sparse_output(kRow * kBatch, 0.0); in TEST() local
1551 matrix_values, ledger, kRow, kCol, vector, kBatch, sparse_output.data()); in TEST()
1553 EXPECT_THAT(sparse_output, in TEST()
1631 std::vector<float> sparse_output(kRow * kBatch, 0.0); in TEST() local
1634 result_scaling_factor, kBatch, sparse_output.data()); in TEST()
1636 EXPECT_THAT(sparse_output, in TEST()
|
| /external/tensorflow/tensorflow/lite/kernels/parse_example/ |
| D | parse_example.cc | 822 const auto* sparse_output = in EvalParseExample() local
825 switch (sparse_output->type) { in EvalParseExample()
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 11889 func DenseToCSRSparseMatrix(scope *Scope, dense_input tf.Output, indices tf.Output) (sparse_output …
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 13810 name: "sparse_output"
|