Searched defs:a_values (Results 1 - 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dsparse_tensor_dense_matmul_op_test.cc26 Node* SparseTensorDenseMatMulNode(Graph* g, Node* a_indices, Node* a_values, argument
32 .Input(a_values)
45 Tensor a_values(DT_FLOAT, TensorShape({nnz}));
51 a_values.flat<float>().setRandom();
66 test::graph::Constant(g, a_values), test::graph::HostConstant(g, a_shape),
H A Dsparse_tensor_dense_matmul_op_gpu.cu.cc34 const T* a_values, const T* b,
55 const T a_value = ldg(a_values + a_ix);
70 typename TTypes<T>::ConstVec a_values,
73 int nnz = a_values.size();
86 nnz, m, b_rows, b_cols, p, a_indices.data(), a_values.data(),
31 SparseTensorDenseMatMulKernel(int nnz, int m, int b_rows, int b_cols, int p, const Tindices* a_indices, const T* a_values, const T* b, T* out) argument
68 Compute(const GPUDevice& d, typename TTypes<T>::Matrix out, typename TTypes<Tindices>::ConstMatrix a_indices, typename TTypes<T>::ConstVec a_values, typename TTypes<T>::ConstMatrix b) argument
H A Dsparse_add_op.cc47 OP_REQUIRES_OK(ctx, ctx->input("a_values", &a_values_t));
57 auto a_values = ctx->input(1).vec<T>(); variable
60 ctx, a_values.size() == a_nnz && b_values.size() == b_nnz,
63 a_values.size(), " and ", b_values.size()));
113 out_values.push_back(a_values(i));
117 s = a_values(i) + b_values(j);
H A Dsparse_sparse_binary_op_shared.cc64 typename TTypes<T>::ConstFlat a_values, int64 a_nnz,
80 a_augmented_values->push_back(a_values(i));
86 a_augmented_values->push_back(a_values(i));
102 a_augmented_values->push_back(a_values(i++));
127 OP_REQUIRES_OK(ctx, ctx->input("a_values", &a_values_t));
146 "Inputs a_values and b_values should be vectors "
153 const auto a_values = a_values_t->vec<T>(); variable
157 ctx, a_values.size() == a_nnz && b_values.size() == b_nnz,
160 a_values.size(), " and ", b_values.size()));
188 UnionSparseIndicesAndValues(a_indices_mat, a_values, a_nn
62 UnionSparseIndicesAndValues( typename TTypes<int64>::ConstMatrix a_indices_mat, typename TTypes<T>::ConstFlat a_values, int64 a_nnz, typename TTypes<int64>::ConstMatrix b_indices_mat, typename TTypes<T>::ConstFlat b_values, int64 b_nnz, int num_dims, std::vector<T> *a_augmented_values, std::vector<T> *b_augmented_values, std::vector<std::pair<bool, int64>> *entries_to_copy) argument
[all...]
H A Dsparse_tensor_dense_add_op.cc35 Status ValidateInputs(const Tensor *a_indices, const Tensor *a_values, argument
42 if (!TensorShapeUtils::IsVector(a_values->shape()) ||
45 "Inputs a_values and a_shape should be vectors "
47 a_values->shape().DebugString(), " and ",
77 OP_REQUIRES_OK(ctx, ctx->input("a_values", &a_values_t));
H A Dsparse_tensor_dense_matmul_op.cc43 const Tensor* a_values; variable
47 OP_REQUIRES_OK(ctx, ctx->input("a_values", &a_values));
62 OP_REQUIRES(ctx, TensorShapeUtils::IsVector(a_values->shape()),
63 errors::InvalidArgument("Tensor 'a_values' is not a vector"));
69 OP_REQUIRES(ctx, nnz == a_values->NumElements(),
71 "match number of entries in a_values"));
109 FastBoundsCheck(a_values->NumElements(), int32max)),
126 if (a_values->NumElements() == 0 || b->NumElements() == 0) {
140 a_indices->matrix<Tindices>(), a_values
247 Compute(const CPUDevice& d, typename TTypes<T>::Matrix out, typename TTypes<Tindices>::ConstMatrix a_indices, typename TTypes<T>::ConstVec a_values, typename TTypes<T>::ConstMatrix b) argument
[all...]
/external/tensorflow/tensorflow/core/platform/
H A Ddevice_tracer_test.cc55 void Initialize(std::initializer_list<float> a_values) { argument
59 test::FillValues<float>(&a_tensor, a_values);
/external/tensorflow/tensorflow/core/common_runtime/
H A Ddirect_session_test.cc59 void Initialize(std::initializer_list<float> a_values) { argument
63 test::FillValues<float>(&a_tensor, a_values);
/external/tensorflow/tensorflow/core/debug/
H A Ddebug_gateway_test.cc53 void Initialize(std::initializer_list<float> a_values) { argument
65 test::FillValues<float>(&a_tensor, a_values);
/external/tensorflow/tensorflow/compiler/xla/tests/
H A Darray_elementwise_ops_test.cc192 std::vector<float> a_values; local
195 a_values.push_back(i / static_cast<float>(count));
199 std::unique_ptr<Literal> a_literal = Literal::CreateR1<float>({a_values});
202 auto a_constant = builder.ConstantR1<float>(a_values);
222 expected.push_back(4 * (a_values[i] + b_values[i]));
/external/tensorflow/tensorflow/tools/graph_transforms/
H A Dquantize_nodes_test.cc208 const std::vector<float>& a_values,
214 test::FillValues<float>(&a_tensor, a_values);
207 TestQuantizeMatMul(int m, int n, int k, const std::vector<float>& a_values, const std::vector<float>& b_values) argument

Completed in 339 milliseconds