/external/tensorflow/tensorflow/core/kernels/ |
D | sparse_tensor_dense_matmul_op.cc | 43 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 …]
|
D | sparse_sparse_binary_op_shared.cc | 64 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()
|
D | sparse_tensor_dense_matmul_op_gpu.cu.cc | 34 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()
|
D | sparse_tensor_dense_matmul_op_test.cc | 26 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()
|
D | sparse_add_op.cc | 57 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()
|
D | sparse_tensor_dense_add_op.cc | 35 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()
|
D | sparse_tensor_dense_matmul_op.h | 34 typename TTypes<T>::ConstVec a_values, typename TTypes<T>::ConstMatrix b);
|
/external/tensorflow/tensorflow/python/kernel_tests/ |
D | sets_test.py | 171 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/ |
D | query_lib.py | 26 (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/ |
D | api_def_SparseSparseMaximum.pbtxt | 11 name: "a_values" 31 counterpart to `a_values` for the other operand; must be of the same dtype.
|
D | api_def_SparseSparseMinimum.pbtxt | 11 name: "a_values" 31 counterpart to `a_values` for the other operand; must be of the same dtype.
|
D | api_def_SparseTensorDenseAdd.pbtxt | 10 name: "a_values"
|
D | api_def_SparseTensorDenseMatMul.pbtxt | 10 name: "a_values"
|
D | api_def_SparseAdd.pbtxt | 10 name: "a_values"
|
/external/tensorflow/tensorflow/python/ops/ |
D | sparse_grad.py | 139 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)
|
D | sparse_ops.py | 1833 a_values=sp_a.values,
|
/external/tensorflow/tensorflow/core/platform/ |
D | device_tracer_test.cc | 55 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/ |
D | debug_gateway_test.cc | 53 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/ |
D | array_elementwise_ops_test.cc | 192 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/ |
D | direct_session_test.cc | 59 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/ |
D | quantize_nodes_test.cc | 208 const std::vector<float>& a_values, in TestQuantizeMatMul() argument 214 test::FillValues<float>(&a_tensor, a_values); in TestQuantizeMatMul()
|
/external/tensorflow/tensorflow/go/op/ |
D | wrappers.go | 12463 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/ |
D | ops_history.v1.pbtxt | 52643 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 …]
|
D | ops_history.v0.pbtxt | 31938 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/ |
D | ops.pbtxt | 24683 name: "a_values" 26939 name: "a_values" 26994 name: "a_values" 27098 name: "a_values" 27156 name: "a_values"
|