Home
last modified time | relevance | path

Searched refs:sparse_values (Results 1 – 25 of 66) sorted by relevance

123

/external/tensorflow/tensorflow/core/kernels/
Dsparse_to_dense_op.cc74 const Tensor& sparse_values = c->input(2); in Compute() local
75 const int64 num_values = sparse_values.NumElements(); in Compute()
77 sparse_values.dims() == 0 || in Compute()
78 (sparse_values.dims() == 1 && num_values == num_elems), in Compute()
80 sparse_values.shape().DebugString(), in Compute()
118 if (TensorShapeUtils::IsScalar(sparse_values.shape())) { in Compute()
122 sparse_values_b_holder->vec<T>().setConstant(sparse_values.scalar<T>()()); in Compute()
124 sparse_values_b = &sparse_values; in Compute()
Dsdca_ops_test.cc175 std::vector<NodeBuilder::NodeOut> sparse_values; in GetGraphs() local
186 sparse_values.reserve(num_sparse_feature_groups); in GetGraphs()
188 sparse_values.push_back( in GetGraphs()
216 .Input(sparse_values) in GetGraphs()
/external/tensorflow/tensorflow/lite/kernels/
Ddensify_test.cc68 std::vector<float> sparse_values = {6, 9, 8, 5, 7}; in TEST() local
76 EXPECT_THAT(m.GetInput(), ElementsAreArray(sparse_values)); in TEST()
82 std::vector<float> sparse_values = {6, 9, 8, 5, 7}; in TEST() local
90 EXPECT_THAT(m.GetInput(), ElementsAreArray(sparse_values)); in TEST()
96 std::vector<int8_t> sparse_values = {6, 9, 8, 5, 7}; in TEST() local
104 EXPECT_THAT(m.GetInput(), ElementsAreArray(sparse_values)); in TEST()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_SparseToDense.pbtxt7 index where `sparse_values[i]` will be placed.
17 name: "sparse_values"
49 dense[i] = (i == sparse_indices ? sparse_values : default_value)
52 dense[sparse_indices[i]] = sparse_values[i]
55 dense[sparse_indices[i][0], ..., sparse_indices[i][d-1]] = sparse_values[i]
58 All other values in `dense` are set to `default_value`. If `sparse_values` is a
Dapi_def_RaggedTensorToSparse.pbtxt17 name: "sparse_values"
39 output=SparseTensor(indices=sparse_indices, values=sparse_values,
Dapi_def_AddSparseToTensorsMap.pbtxt10 name: "sparse_values"
44 `sparse_values`, and `sparse_shape`.
Dapi_def_AddManySparseToTensorsMap.pbtxt11 name: "sparse_values"
46 `sparse_values`, and `sparse_shape`, where
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dsparse_to_dense_op.cc67 xla::XlaOp sparse_values = context->Input(2); in Compile() local
71 sparse_values = Broadcast(sparse_values, {num_elems}); in Compile()
76 auto result = XlaScatter(buffer, sparse_values, indices, in Compile()
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DSerializeManySparse.pbtxt8 name: "sparse_values"
31 name: "sparse_values"
DSerializeSparse.pbtxt8 name: "sparse_values"
31 name: "sparse_values"
DDeserializeSparse.pbtxt12 name: "sparse_values"
35 name: "sparse_values"
DRaggedTensorToSparse.pbtxt17 name: "sparse_values"
51 name: "sparse_values"
DDeserializeManySparse.pbtxt12 name: "sparse_values"
DTakeManySparseFromTensorsMap.pbtxt12 name: "sparse_values"
DAddSparseToTensorsMap.pbtxt8 name: "sparse_values"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DDeserializeSparse.pbtxt12 name: "sparse_values"
35 name: "sparse_values"
DSerializeManySparse.pbtxt8 name: "sparse_values"
31 name: "sparse_values"
DSerializeSparse.pbtxt8 name: "sparse_values"
31 name: "sparse_values"
DRaggedTensorToSparse.pbtxt17 name: "sparse_values"
51 name: "sparse_values"
DDeserializeManySparse.pbtxt12 name: "sparse_values"
DAddSparseToTensorsMap.pbtxt8 name: "sparse_values"
DAddManySparseToTensorsMap.pbtxt8 name: "sparse_values"
DTakeManySparseFromTensorsMap.pbtxt12 name: "sparse_values"
/external/tensorflow/tensorflow/python/distribute/
Dcross_device_ops.py861 dense_values, dense_indices, sparse_values, sparse_indices = (
867 if sparse_values:
869 sparse_values)
906 def _do_batch_all_reduce_sparse(self, reduce_op, sparse_values): argument
911 len(sparse_values), 10)
915 reduce_op, zip(sparse_values, sparse_values))
1183 dense_values, dense_indices, sparse_values, sparse_indices = (
1190 if sparse_values:
1192 sparse_values, options)
/external/tensorflow/tensorflow/compiler/tests/
Dsparse_to_dense_op_test.py33 sparse_values, argument
41 sparse_values,

123