| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_tensor_dense_matmul_op.cc | 44 const Tensor* a_values; in Compute() local
48 OP_REQUIRES_OK(ctx, ctx->input("a_values", &a_values)); in Compute()
63 OP_REQUIRES(ctx, TensorShapeUtils::IsVector(a_values->shape()), in Compute()
70 OP_REQUIRES(ctx, nnz == a_values->NumElements(), in Compute()
110 FastBoundsCheck(a_values->NumElements(), int32max)), in Compute()
127 if (a_values->NumElements() == 0 || b->NumElements() == 0) { in Compute()
141 a_indices->matrix<Tindices>(), a_values->vec<T>(), \ in Compute()
188 typename TTypes<T>::ConstVec a_values, \
251 typename TTypes<T>::ConstVec a_values, in Compute()
253 const std::size_t nnz = a_values.size(); in Compute()
[all …]
|
| D | sparse_sparse_binary_op_shared.cc | 64 typename TTypes<T>::ConstFlat a_values, int64 a_nnz, in UnionSparseIndicesAndValues() argument
80 a_augmented_values->push_back(a_values(i)); in UnionSparseIndicesAndValues()
86 a_augmented_values->push_back(a_values(i)); in UnionSparseIndicesAndValues()
102 a_augmented_values->push_back(a_values(i++)); in UnionSparseIndicesAndValues()
153 const auto a_values = a_values_t->vec<T>(); in Compute() local
157 ctx, a_values.size() == a_nnz && b_values.size() == b_nnz, in Compute()
160 a_values.size(), " and ", b_values.size())); in Compute()
188 UnionSparseIndicesAndValues(a_indices_mat, a_values, a_nnz, b_indices_mat, in Compute()
|
| D | sparse_tensor_dense_matmul_op_gpu.cu.cc | 32 const Tindices* __restrict__ a_indices, const T* __restrict__ a_values, in SparseTensorDenseMatMulKernel() argument
53 const T a_value = ldg(a_values + a_ix); in SparseTensorDenseMatMulKernel()
68 typename TTypes<T>::ConstVec a_values, in Compute()
71 int nnz = a_values.size(); in Compute()
85 b_rows, b_cols, p, a_indices.data(), a_values.data(), b.data(), in Compute()
|
| D | sparse_tensor_dense_matmul_op_test.cc | 26 Node* SparseTensorDenseMatMulNode(Graph* g, Node* a_indices, Node* a_values, in SparseTensorDenseMatMulNode() argument
32 .Input(a_values) in SparseTensorDenseMatMulNode()
45 Tensor a_values(DT_FLOAT, TensorShape({nnz})); in SparseTensorDenseMatmul() local
51 a_values.flat<float>().setRandom(); in SparseTensorDenseMatmul()
66 test::graph::Constant(g, a_values), test::graph::HostConstant(g, a_shape), in SparseTensorDenseMatmul()
|
| D | sparse_add_op.cc | 57 auto a_values = ctx->input(1).vec<T>(); in Compute() local
60 ctx, a_values.size() == a_nnz && b_values.size() == b_nnz, in Compute()
63 a_values.size(), " and ", b_values.size())); in Compute()
113 out_values.push_back(a_values(i)); in Compute()
117 s = a_values(i) + b_values(j); in Compute()
|
| D | sparse_tensor_dense_add_op.cc | 35 Status ValidateInputs(const Tensor *a_indices, const Tensor *a_values, in ValidateInputs() argument
42 if (!TensorShapeUtils::IsVector(a_values->shape()) || in ValidateInputs()
47 a_values->shape().DebugString(), " and ", in ValidateInputs()
|
| /external/autotest/tko/ |
| D | query_lib.py | 26 (a_sql, a_values) = me(match.group(1), valid_field_dict)
29 a_values + b_values)
34 (a_sql, a_values) = me(match.group(1), valid_field_dict)
37 a_values + b_values)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sets_test.py | 175 a_values = [[9, 1, 5], [2, 4, 3]]
183 a = _constant(a_values, dtype=dtype)
195 sp_a = _dense_to_sparse(a_values, dtype=dtype)
212 a_values = [[9, 1, 5], [2, 4, 3]]
220 a = _constant(a_values, dtype)
237 a_values = [[1, 1, 3]]
245 a = _constant(a_values, dtype=dtype)
268 sp_a = _dense_to_sparse(a_values, dtype=dtype)
552 a_values = [[1, 1, 1], [1, 5, 9], [4, 5, 3], [5, 5, 1]]
563 a = _constant(a_values, dtype=dtype)
[all …]
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | SparseSparseMaximum.pbtxt | 8 name: "a_values"
60 name: "a_values"
114 name: "a_values"
169 name: "a_values"
|
| D | SparseSparseMinimum.pbtxt | 8 name: "a_values"
65 name: "a_values"
124 name: "a_values"
184 name: "a_values"
|
| D | SparseTensorDenseAdd.pbtxt | 8 name: "a_values"
63 name: "a_values"
120 name: "a_values"
178 name: "a_values"
|
| D | SparseTensorDenseMatMul.pbtxt | 8 name: "a_values"
49 name: "a_values"
|
| D | SparseAdd.pbtxt | 8 name: "a_values"
90 name: "a_values"
176 name: "a_values"
264 name: "a_values"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | SparseSparseMaximum.pbtxt | 8 name: "a_values"
60 name: "a_values"
114 name: "a_values"
169 name: "a_values"
|
| D | SparseTensorDenseAdd.pbtxt | 8 name: "a_values"
63 name: "a_values"
120 name: "a_values"
178 name: "a_values"
|
| D | SparseSparseMinimum.pbtxt | 8 name: "a_values"
65 name: "a_values"
124 name: "a_values"
184 name: "a_values"
|
| D | SparseTensorDenseMatMul.pbtxt | 8 name: "a_values"
49 name: "a_values"
|
| D | SparseAdd.pbtxt | 8 name: "a_values"
90 name: "a_values"
176 name: "a_values"
264 name: "a_values"
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/sparse/ |
| D | csr_sparse_matrix_ops_test.py | 108 a_values = np.asarray([1.0, 5.0], dtype=np.float32)
110 a_sparse_mat = sparse.coo_matrix((a_values,
124 values_proto = tensor_util.make_tensor_proto(a_values)
143 a_values = [1.0, 5.0, -1.0, -2.0]
145 a_sparse_mat = sparse.coo_matrix((a_values,
151 a_st = sparse_tensor.SparseTensor(a_indices, a_values, a_dense_shape)
166 self.assertAllClose(a_values, a_sm_values_values)
174 self.assertAllClose(a_values, a_st_rt_value.values)
208 a_values = [1.0, 5.0, 6.0]
219 a_st = sparse_tensor.SparseTensor(a_indices, a_values, a_dense_shape)
[all …]
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_SparseSparseMinimum.pbtxt | 11 name: "a_values"
31 counterpart to `a_values` for the other operand; must be of the same dtype.
|
| D | api_def_SparseSparseMaximum.pbtxt | 11 name: "a_values"
31 counterpart to `a_values` for the other operand; must be of the same dtype.
|
| /external/python/rsa/rsa/ |
| D | common.py | 147 def crt(a_values: typing.Iterable[int], modulo_values: typing.Iterable[int]) -> int:
173 for (m_i, a_i) in zip(modulo_values, a_values):
|
| /external/tensorflow/tensorflow/lite/micro/memory_planner/ |
| D | greedy_memory_planner_test.cc | 38 int a_values[a_size] = {10, 9, 8, 7, 6, 5, 4, 3, 2, 1}; in TF_LITE_MICRO_TEST() local
42 tflite::ReverseSortInPlace(a_values, a_ids, a_size); in TF_LITE_MICRO_TEST()
44 TF_LITE_MICRO_EXPECT_EQ(a_expected_values[i], a_values[i]); in TF_LITE_MICRO_TEST()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | sparse_grad.py | 166 a_indices, a_values, a_shape = op.inputs[:3]
171 a_type = a_values.dtype.base_dtype
182 a_indices, a_values, a_shape, grad, adjoint_a=not adj_a)
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | mul_op.cc | 138 auto a_values = a.values().flat<T>(); in Compute() local
147 d, c_values, a_values, b, error_ptr); in Compute()
|