| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_to_dense_op.cc | 52 const Tensor& sparse_values, in CheckSparseToDenseShapes() argument
77 const int64_t num_values = sparse_values.NumElements(); in CheckSparseToDenseShapes()
78 if (sparse_values.dims() != 0 && in CheckSparseToDenseShapes()
79 (sparse_values.dims() != 1 || num_values != num_elems)) { in CheckSparseToDenseShapes()
81 sparse_values.shape().DebugString(), in CheckSparseToDenseShapes()
106 const Tensor& sparse_values = c->input(2); in Compute() local
109 sparse_values, default_value)); in Compute()
144 if (TensorShapeUtils::IsScalar(sparse_values.shape())) { in Compute()
148 sparse_values_b_holder->vec<T>().setConstant(sparse_values.scalar<T>()()); in Compute()
150 sparse_values_b = &sparse_values; in Compute()
[all …]
|
| D | sdca_ops_test.cc | 176 std::vector<NodeBuilder::NodeOut> sparse_values; in GetGraphs() local
187 sparse_values.reserve(num_sparse_feature_groups); in GetGraphs()
189 sparse_values.push_back( in GetGraphs()
217 .Input(sparse_values) in GetGraphs()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | densify_test.cc | 70 std::vector<float> sparse_values = {6, 9, 8, 5, 7}; in TEST() local
78 EXPECT_THAT(m.GetInput(), ElementsAreArray(sparse_values)); in TEST()
84 std::vector<float> sparse_values = {6, 9, 8, 5, 7}; in TEST() local
92 EXPECT_THAT(m.GetInput(), ElementsAreArray(sparse_values)); in TEST()
98 std::vector<int8_t> sparse_values = {6, 9, 8, 5, 7}; in TEST() local
106 EXPECT_THAT(m.GetInput(), ElementsAreArray(sparse_values)); in TEST()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_SparseToDense.pbtxt | 7 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
|
| D | api_def_RaggedTensorToSparse.pbtxt | 17 name: "sparse_values"
39 output=SparseTensor(indices=sparse_indices, values=sparse_values,
|
| D | api_def_AddSparseToTensorsMap.pbtxt | 10 name: "sparse_values"
44 `sparse_values`, and `sparse_shape`.
|
| D | api_def_AddManySparseToTensorsMap.pbtxt | 11 name: "sparse_values"
46 `sparse_values`, and `sparse_shape`, where
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | sparse_to_dense_op.cc | 69 xla::XlaOp sparse_values = context->Input(2); in Compile() local
73 sparse_values = Broadcast(sparse_values, {num_elems}); in Compile()
91 auto result = XlaScatter(buffer, sparse_values, indices, in Compile()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | DeserializeSparse.pbtxt | 12 name: "sparse_values"
35 name: "sparse_values"
|
| D | SerializeManySparse.pbtxt | 8 name: "sparse_values"
31 name: "sparse_values"
|
| D | SerializeSparse.pbtxt | 8 name: "sparse_values"
31 name: "sparse_values"
|
| D | RaggedTensorToSparse.pbtxt | 17 name: "sparse_values"
51 name: "sparse_values"
|
| D | DeserializeManySparse.pbtxt | 12 name: "sparse_values"
|
| D | TakeManySparseFromTensorsMap.pbtxt | 12 name: "sparse_values"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | DeserializeSparse.pbtxt | 12 name: "sparse_values"
35 name: "sparse_values"
|
| D | SerializeSparse.pbtxt | 8 name: "sparse_values"
31 name: "sparse_values"
|
| D | SerializeManySparse.pbtxt | 8 name: "sparse_values"
31 name: "sparse_values"
|
| D | RaggedTensorToSparse.pbtxt | 17 name: "sparse_values"
51 name: "sparse_values"
|
| D | DeserializeManySparse.pbtxt | 12 name: "sparse_values"
|
| D | AddSparseToTensorsMap.pbtxt | 8 name: "sparse_values"
|
| /external/tensorflow/tensorflow/security/advisory/ |
| D | tfsa-2021-021.md | 16 sparse_values = []
30 sparse_values=sparse_values,
|
| D | tfsa-2021-012.md | 15 sparse_values = tf.ones([1], dtype=tf.int64)
21 sparse_values=sparse_values,
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_ops.py | 884 dense_values, dense_indices, sparse_values, sparse_indices = (
890 if sparse_values:
892 sparse_values)
929 def _do_batch_all_reduce_sparse(self, reduce_op, sparse_values): argument
934 len(sparse_values), 10)
938 reduce_op, zip(sparse_values, sparse_values))
1130 dense_values, dense_indices, sparse_values, sparse_indices = (
1165 if sparse_values:
1169 "%d, group_size = %d, implementation = %s", len(sparse_values),
1172 for indexed_slice in sparse_values:
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_tensor_dense_matmul_op_deterministic_test.py | 62 sparse_values = maybe_complex(
66 sparse_input = sparse_tensor.SparseTensor(indices, sparse_values,
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | sparse_to_dense_op_test.py | 33 sparse_values, argument
41 sparse_values,
|