/external/tensorflow/tensorflow/core/kernels/ |
H A D | sparse_tensor_dense_matmul_op_test.cc | 26 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 D | sparse_tensor_dense_matmul_op_gpu.cu.cc | 34 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 D | sparse_add_op.cc | 47 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 D | sparse_sparse_binary_op_shared.cc | 64 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 D | sparse_tensor_dense_add_op.cc | 35 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 D | sparse_tensor_dense_matmul_op.cc | 43 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 D | device_tracer_test.cc | 55 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 D | direct_session_test.cc | 59 void Initialize(std::initializer_list<float> a_values) { argument 63 test::FillValues<float>(&a_tensor, a_values);
|
/external/tensorflow/tensorflow/core/debug/ |
H A D | debug_gateway_test.cc | 53 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 D | array_elementwise_ops_test.cc | 192 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 D | quantize_nodes_test.cc | 208 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
|