/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...] |