| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_tensor_dense_matmul_op.cc | 43 const Tensor* a_values; in Compute() local
47 OP_REQUIRES_OK(ctx, ctx->input("a_values", &a_values)); in Compute()
62 OP_REQUIRES(ctx, TensorShapeUtils::IsVector(a_values->shape()), in Compute()
69 OP_REQUIRES(ctx, nnz == a_values->NumElements(), in Compute()
109 FastBoundsCheck(a_values->NumElements(), int32max)), in Compute()
126 if (a_values->NumElements() == 0 || b->NumElements() == 0) { in Compute()
140 a_indices->matrix<Tindices>(), a_values->vec<T>(), \ in Compute()
186 typename TTypes<T>::ConstVec a_values, \
249 typename TTypes<T>::ConstVec a_values, in Compute()
251 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 | 34 const T* a_values, const T* b, in SparseTensorDenseMatMulKernel() argument
55 const T a_value = ldg(a_values + a_ix); in SparseTensorDenseMatMulKernel()
70 typename TTypes<T>::ConstVec a_values, in Compute()
73 int nnz = a_values.size(); in Compute()
87 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()
|
| D | sparse_tensor_dense_matmul_op.h | 34 typename TTypes<T>::ConstVec a_values, typename TTypes<T>::ConstMatrix b);
|
| /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/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/python/rsa/rsa/ |
| D | common.py | 150 def crt(a_values, modulo_values): argument
176 for (m_i, a_i) in zip(modulo_values, a_values):
|
| /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.
|
| D | api_def_SparseTensorDenseAdd.pbtxt | 10 name: "a_values"
|
| D | api_def_SparseTensorDenseMatMul.pbtxt | 10 name: "a_values"
|
| D | api_def_SparseAdd.pbtxt | 10 name: "a_values"
|
| /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)
|
| D | sparse_ops.py | 2341 a_values=sp_a.values,
|
| /external/tensorflow/tensorflow/core/platform/ |
| D | device_tracer_test.cc | 55 void Initialize(std::initializer_list<float> a_values) { in Initialize() argument
59 test::FillValues<float>(&a_tensor, a_values); in Initialize()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | array_elementwise_ops_test.cc | 315 std::vector<float> a_values; in TEST_P() local
318 a_values.push_back(i / static_cast<float>(count)); in TEST_P()
322 Literal a_literal = LiteralUtil::CreateR1<float>({a_values}); in TEST_P()
325 auto a_constant = ConstantR1<float>(&builder, a_values); in TEST_P()
345 expected.push_back(4 * (a_values[i] + b_values[i])); in TEST_P()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | quantize_nodes_test.cc | 208 const std::vector<float>& a_values, in TestQuantizeMatMul() argument
214 test::FillValues<float>(&a_tensor, a_values); in TestQuantizeMatMul()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | direct_session_test.cc | 84 void Initialize(std::initializer_list<float> a_values) { in Initialize() argument
88 test::FillValues<float>(&a_tensor, a_values); in Initialize()
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 3417 …argspec: "args=[\'a_indices\', \'a_values\', \'a_shape\', \'b_indices\', \'b_values\', \'b_shape\'…
3573 …argspec: "args=[\'a_indices\', \'a_values\', \'a_shape\', \'b_indices\', \'b_values\', \'b_shape\'…
3577 …argspec: "args=[\'a_indices\', \'a_values\', \'a_shape\', \'b_indices\', \'b_values\', \'b_shape\'…
3585 …argspec: "args=[\'a_indices\', \'a_values\', \'a_shape\', \'b\', \'name\'], varargs=None, keywords…
3589 …argspec: "args=[\'a_indices\', \'a_values\', \'a_shape\', \'b\', \'adjoint_a\', \'adjoint_b\', \'n…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 3417 …argspec: "args=[\'a_indices\', \'a_values\', \'a_shape\', \'b_indices\', \'b_values\', \'b_shape\'…
3573 …argspec: "args=[\'a_indices\', \'a_values\', \'a_shape\', \'b_indices\', \'b_values\', \'b_shape\'…
3577 …argspec: "args=[\'a_indices\', \'a_values\', \'a_shape\', \'b_indices\', \'b_values\', \'b_shape\'…
3585 …argspec: "args=[\'a_indices\', \'a_values\', \'a_shape\', \'b\', \'name\'], varargs=None, keywords…
3589 …argspec: "args=[\'a_indices\', \'a_values\', \'a_shape\', \'b\', \'adjoint_a\', \'adjoint_b\', \'n…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 12996 func SparseTensorDenseAdd(scope *Scope, a_indices tf.Output, a_values tf.Output, a_shape tf.Output,…
13003 a_indices, a_values, a_shape, b,
13947 func SparseTensorDenseMatMul(scope *Scope, a_indices tf.Output, a_values tf.Output, a_shape tf.Outp…
13958 a_indices, a_values, a_shape, b,
13991 func SparseAdd(scope *Scope, a_indices tf.Output, a_values tf.Output, a_shape tf.Output, b_indices …
13998 a_indices, a_values, a_shape, b_indices, b_values, b_shape, thresh,
15292 func SparseSparseMinimum(scope *Scope, a_indices tf.Output, a_values tf.Output, a_shape tf.Output, …
15299 a_indices, a_values, a_shape, b_indices, b_values, b_shape,
28165 func SparseSparseMaximum(scope *Scope, a_indices tf.Output, a_values tf.Output, a_shape tf.Output, …
28172 a_indices, a_values, a_shape, b_indices, b_values, b_shape,
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v1.pbtxt | 71787 name: "a_values"
71869 name: "a_values"
71955 name: "a_values"
72043 name: "a_values"
78648 name: "a_values"
78700 name: "a_values"
78754 name: "a_values"
78809 name: "a_values"
78864 name: "a_values"
78921 name: "a_values"
[all …]
|