Home
last modified time | relevance | path

Searched refs:b_values (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsparse_sparse_binary_op_shared.cc66 typename TTypes<T>::ConstFlat b_values, int64 b_nnz, int num_dims, in UnionSparseIndicesAndValues() argument
87 b_augmented_values->push_back(b_values(j)); in UnionSparseIndicesAndValues()
94 b_augmented_values->push_back(b_values(j)); in UnionSparseIndicesAndValues()
108 b_augmented_values->push_back(b_values(j++)); in UnionSparseIndicesAndValues()
154 const auto b_values = b_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()
189 b_values, b_nnz, num_dims, &a_augmented_values, in Compute()
Dsparse_add_op.cc58 auto b_values = ctx->input(4).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()
117 s = a_values(i) + b_values(j); in Compute()
127 out_values.push_back(b_values(j)); in Compute()
Dctc_loss_op.cc122 std::vector<int>* b_values = &labels_t[batch_indices]; in Compute() local
123 b_values->resize(values.size()); in Compute()
124 for (int i = 0; i < values.size(); ++i) (*b_values)[i] = values(i); in Compute()
/external/autotest/tko/
Dquery_lib.py27 (b_sql, b_values) = me(match.group(2), valid_field_dict)
29 a_values + b_values)
35 (b_sql, b_values) = me(match.group(2), valid_field_dict)
37 a_values + b_values)
/external/tensorflow/tensorflow/python/kernel_tests/
Dsets_test.py176 b_values = [[1, 9], [1]]
184 sp_b = _dense_to_sparse(b_values, dtype=dtype)
213 b_values = [[1, 9], [1, 5]]
221 b = _constant(b_values, dtype)
238 b_values = [[1]]
246 b = _constant(b_values, dtype=dtype)
257 sp_b = _dense_to_sparse(b_values, dtype=dtype)
553 b_values = [[], [1, 2], [1, 2, 2], []]
564 sp_b = _dense_to_sparse(b_values, dtype=dtype)
622 b_values = [[1, 2, 6], [1, 2, 2]]
[all …]
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_SparseSparseMinimum.pbtxt29 name: "b_values"
Dapi_def_SparseSparseMaximum.pbtxt29 name: "b_values"
Dapi_def_SparseAdd.pbtxt28 name: "b_values"
/external/tensorflow/tensorflow/compiler/xla/tests/
Darray_elementwise_ops_test.cc316 std::vector<float> b_values; in TEST_P() local
319 b_values.push_back(2 * i / static_cast<float>(count + 2)); in TEST_P()
328 Literal b_literal = LiteralUtil::CreateR1<float>({b_values}); in TEST_P()
332 auto b_constant = ConstantR1<float>(&builder, b_values); in TEST_P()
345 expected.push_back(4 * (a_values[i] + b_values[i])); in TEST_P()
/external/tensorflow/tensorflow/tools/graph_transforms/
Dquantize_nodes_test.cc209 const std::vector<float>& b_values) { in TestQuantizeMatMul() argument
218 test::FillValues<float>(&b_tensor, b_values); in TestQuantizeMatMul()
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.raw_ops.pbtxt3417 …argspec: "args=[\'a_indices\', \'a_values\', \'a_shape\', \'b_indices\', \'b_values\', \'b_shape\'…
3573 …argspec: "args=[\'a_indices\', \'a_values\', \'a_shape\', \'b_indices\', \'b_values\', \'b_shape\'…
3577 …argspec: "args=[\'a_indices\', \'a_values\', \'a_shape\', \'b_indices\', \'b_values\', \'b_shape\'…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.raw_ops.pbtxt3417 …argspec: "args=[\'a_indices\', \'a_values\', \'a_shape\', \'b_indices\', \'b_values\', \'b_shape\'…
3573 …argspec: "args=[\'a_indices\', \'a_values\', \'a_shape\', \'b_indices\', \'b_values\', \'b_shape\'…
3577 …argspec: "args=[\'a_indices\', \'a_values\', \'a_shape\', \'b_indices\', \'b_values\', \'b_shape\'…
/external/tensorflow/tensorflow/go/op/
Dwrappers.go13991 … tf.Output, a_values tf.Output, a_shape tf.Output, b_indices tf.Output, b_values tf.Output, b_shap…
13998 a_indices, a_values, a_shape, b_indices, b_values, b_shape, thresh,
15292 … tf.Output, a_values tf.Output, a_shape tf.Output, b_indices tf.Output, b_values tf.Output, b_shap…
15299 a_indices, a_values, a_shape, b_indices, b_values, b_shape,
28165 … tf.Output, a_values tf.Output, a_shape tf.Output, b_indices tf.Output, b_values tf.Output, b_shap…
28172 a_indices, a_values, a_shape, b_indices, b_values, b_shape,
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v1.pbtxt71799 name: "b_values"
71881 name: "b_values"
71967 name: "b_values"
72055 name: "b_values"
78660 name: "b_values"
78712 name: "b_values"
78766 name: "b_values"
78821 name: "b_values"
78876 name: "b_values"
78933 name: "b_values"
[all …]
Dops_history.v2.pbtxt71172 name: "b_values"
71254 name: "b_values"
71340 name: "b_values"
71428 name: "b_values"
78033 name: "b_values"
78085 name: "b_values"
78139 name: "b_values"
78194 name: "b_values"
78249 name: "b_values"
78306 name: "b_values"
[all …]
Dops_history.v0.pbtxt31950 name: "b_values"
32030 name: "b_values"
34690 name: "b_values"
34742 name: "b_values"
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt35187 name: "b_values"
37511 name: "b_values"
37566 name: "b_values"