Home
last modified time | relevance | path

Searched refs:a_values (Results 1 – 25 of 25) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsparse_tensor_dense_matmul_op.cc43 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 …]
Dsparse_sparse_binary_op_shared.cc64 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()
Dsparse_tensor_dense_matmul_op_gpu.cu.cc34 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()
86 nnz, m, b_rows, b_cols, p, a_indices.data(), a_values.data(), in Compute()
Dsparse_tensor_dense_matmul_op_test.cc26 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()
Dsparse_add_op.cc57 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()
Dsparse_tensor_dense_add_op.cc35 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()
Dsparse_tensor_dense_matmul_op.h34 typename TTypes<T>::ConstVec a_values, typename TTypes<T>::ConstMatrix b);
/external/tensorflow/tensorflow/python/kernel_tests/
Dsets_test.py171 a_values = [[9, 1, 5], [2, 4, 3]]
179 a = _constant(a_values, dtype=dtype)
191 sp_a = _dense_to_sparse(a_values, dtype=dtype)
207 a_values = [[9, 1, 5], [2, 4, 3]]
215 a = _constant(a_values, dtype)
231 a_values = [[1, 1, 3]]
239 a = _constant(a_values, dtype=dtype)
262 sp_a = _dense_to_sparse(a_values, dtype=dtype)
544 a_values = [[1, 1, 1], [1, 5, 9], [4, 5, 3], [5, 5, 1]]
555 a = _constant(a_values, dtype=dtype)
[all …]
/external/autotest/tko/
Dquery_lib.py26 (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/core/api_def/base_api/
Dapi_def_SparseSparseMaximum.pbtxt11 name: "a_values"
31 counterpart to `a_values` for the other operand; must be of the same dtype.
Dapi_def_SparseSparseMinimum.pbtxt11 name: "a_values"
31 counterpart to `a_values` for the other operand; must be of the same dtype.
Dapi_def_SparseTensorDenseAdd.pbtxt10 name: "a_values"
Dapi_def_SparseTensorDenseMatMul.pbtxt10 name: "a_values"
Dapi_def_SparseAdd.pbtxt10 name: "a_values"
/external/tensorflow/tensorflow/python/ops/
Dsparse_grad.py139 a_indices, a_values, a_shape = op.inputs[:3]
144 a_type = a_values.dtype.base_dtype
155 a_indices, a_values, a_shape, grad, adjoint_a=not adj_a)
Dsparse_ops.py1833 a_values=sp_a.values,
/external/tensorflow/tensorflow/core/platform/
Ddevice_tracer_test.cc55 void Initialize(std::initializer_list<float> a_values) { in Initialize() argument
59 test::FillValues<float>(&a_tensor, a_values); in Initialize()
/external/tensorflow/tensorflow/core/debug/
Ddebug_gateway_test.cc53 void Initialize(std::initializer_list<float> a_values) { in Initialize() argument
65 test::FillValues<float>(&a_tensor, a_values); in Initialize()
/external/tensorflow/tensorflow/compiler/xla/tests/
Darray_elementwise_ops_test.cc192 std::vector<float> a_values; in TEST_P() local
195 a_values.push_back(i / static_cast<float>(count)); in TEST_P()
199 std::unique_ptr<Literal> a_literal = Literal::CreateR1<float>({a_values}); in TEST_P()
202 auto a_constant = builder.ConstantR1<float>(a_values); in TEST_P()
222 expected.push_back(4 * (a_values[i] + b_values[i])); in TEST_P()
/external/tensorflow/tensorflow/core/common_runtime/
Ddirect_session_test.cc59 void Initialize(std::initializer_list<float> a_values) { in Initialize() argument
63 test::FillValues<float>(&a_tensor, a_values); in Initialize()
/external/tensorflow/tensorflow/tools/graph_transforms/
Dquantize_nodes_test.cc208 const std::vector<float>& a_values, in TestQuantizeMatMul() argument
214 test::FillValues<float>(&a_tensor, a_values); in TestQuantizeMatMul()
/external/tensorflow/tensorflow/go/op/
Dwrappers.go12463 func SparseTensorDenseMatMul(scope *Scope, a_indices tf.Output, a_values tf.Output, a_shape tf.Outp…
12474 a_indices, a_values, a_shape, b,
12857 func SparseTensorDenseAdd(scope *Scope, a_indices tf.Output, a_values tf.Output, a_shape tf.Output,…
12864 a_indices, a_values, a_shape, b,
13017 func SparseAdd(scope *Scope, a_indices tf.Output, a_values tf.Output, a_shape tf.Output, b_indices …
13024 a_indices, a_values, a_shape, b_indices, b_values, b_shape, thresh,
13305 func SparseSparseMinimum(scope *Scope, a_indices tf.Output, a_values tf.Output, a_shape tf.Output, …
13312 a_indices, a_values, a_shape, b_indices, b_values, b_shape,
26284 func SparseSparseMaximum(scope *Scope, a_indices tf.Output, a_values tf.Output, a_shape tf.Output, …
26291 a_indices, a_values, a_shape, b_indices, b_values, b_shape,
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v1.pbtxt52643 name: "a_values"
52725 name: "a_values"
52811 name: "a_values"
52899 name: "a_values"
59312 name: "a_values"
59364 name: "a_values"
59418 name: "a_values"
59473 name: "a_values"
59528 name: "a_values"
59585 name: "a_values"
[all …]
Dops_history.v0.pbtxt31938 name: "a_values"
32018 name: "a_values"
34678 name: "a_values"
34730 name: "a_values"
34831 name: "a_values"
34886 name: "a_values"
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt24683 name: "a_values"
26939 name: "a_values"
26994 name: "a_values"
27098 name: "a_values"
27156 name: "a_values"