| /external/tensorflow/tensorflow/core/kernels/ |
| 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);
|
| 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_nnz, b_indices_mat
[all...] |
| 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->vec<T>(),
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| array_elementwise_ops_test.cc | 315 std::vector<float> a_values; local
318 a_values.push_back(i / static_cast<float>(count));
322 Literal a_literal = LiteralUtil::CreateR1<float>({a_values});
325 auto a_constant = ConstantR1<float>(&builder, a_values);
345 expected.push_back(4 * (a_values[i] + b_values[i]));
[all...] |
