| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | densify_test.cc | 70 std::vector<float> sparse_values = {6, 9, 8, 5, 7}; in TEST() local
84 std::vector<float> sparse_values = {6, 9, 8, 5, 7}; in TEST() local
98 std::vector<int8_t> sparse_values = {6, 9, 8, 5, 7}; in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | sparse_to_dense_op.cc | 52 const Tensor& sparse_values, in CheckSparseToDenseShapes()
106 const Tensor& sparse_values = c->input(2); in Compute() local
213 const Tensor& sparse_values = c->input(2); in ComputeAsync() local
|
| H A D | sparse_to_dense_op_test.cc | 217 Tensor sparse_values(DT_FLOAT, TensorShape({N})); in BM_SparseToDense() local
|
| H A D | example_parsing_ops.cc | 259 OpOutputList sparse_values; in WriteOutput() local
371 OpOutputList sparse_values; in Compute() local
|
| H A D | sdca_ops_test.cc | 176 std::vector<NodeBuilder::NodeOut> sparse_values; in GetGraphs() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | sparse_to_dense_op.cc | 69 xla::XlaOp sparse_values = context->Input(2); in Compile() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | sparse_to_dense_op_test.py | 29 sparse_values, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | sparse_ops_test.py | 125 def f(sparse_values, default_value): argument
|
| H A D | sparse_ops.py | 1176 sparse_values, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | example_proto_fast_parsing.h | 120 std::vector<Tensor> sparse_values; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | cross_device_ops.py | 926 def _do_batch_all_reduce_sparse(self, reduce_op, sparse_values): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/parse_example/ |
| H A D | parse_example.cc | 67 std::vector<TfLiteTensor*> sparse_values; member
|
| /aosp_15_r20/external/pytorch/test/cpp/c10d/ |
| H A D | ProcessGroupNCCLTest.cpp | 191 at::Tensor sparse_values = tensor.index_select( in to_sparse_row_indices_format() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/ |
| H A D | SparseTensorMath.cpp | 2025 Tensor sparse_values = values.slice(0, mat_el_begin_idx, mat_el_end_idx); in bmm_out_sparse_cpu() local
|