| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_to_dense_op.cc | 74 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()
110 if (TensorShapeUtils::IsScalar(sparse_values.shape())) { in Compute()
114 sparse_values_b.vec<T>().setConstant(sparse_values.scalar<T>()()); in Compute()
116 sparse_values_b = sparse_values; in Compute()
|
| D | sdca_ops_test.cc | 175 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()
|
| D | example_parsing_ops.cc | 148 OpOutputList sparse_values; in Compute() local
152 OP_REQUIRES_OK(ctx, ctx->output_list("sparse_values", &sparse_values)); in Compute()
159 sparse_values.set(d, result.sparse_values[d]); in Compute()
244 OpOutputList sparse_values; in Compute() local
248 OP_REQUIRES_OK(ctx, ctx->output_list("sparse_values", &sparse_values)); in Compute()
255 sparse_values.set(d, result.sparse_values[d]); in Compute()
409 context_sparse_values.set(d, context_result.sparse_values[d]); in Compute()
414 feature_list_sparse_values.set(d, feature_list_result.sparse_values[d]); in Compute()
|
| D | sparse_to_dense_op_test.cc | 219 Tensor sparse_values(DT_FLOAT, TensorShape({N})); in BM_SparseToDense() local
233 {&sparse_indices, &output_shape, &sparse_values, &default_value}) { in BM_SparseToDense()
|
| /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
|
| D | api_def_SerializeSparse.pbtxt | 10 name: "sparse_values"
|
| D | api_def_SerializeManySparse.pbtxt | 10 name: "sparse_values"
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_to_dense_op_py_test.py | 32 sparse_values, argument
38 sparse_values,
141 sparse_values=[-1.0, 1.0],
151 sparse_values=[-1.0, 1.0],
162 sparse_values=[-1.0, 1.0],
170 sparse_values=[-1.0, 1.0],
181 sparse_values=[-1.0, 1.0],
189 sparse_values=[-1.0, 1.0],
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | sparse_to_dense_op.cc | 67 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/contrib/tensor_forest/python/ |
| D | tensor_forest.py | 614 sparse_values = []
618 sparse_values = sparse_features.values
628 sparse_values,
645 sparse_values,
674 sparse_values = []
678 sparse_values = sparse_features.values
687 sparse_values,
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | embedding_ops.py | 381 sparse_values, argument
422 if not isinstance(sparse_values, sparse_tensor.SparseTensor):
426 params + [sparse_values]) as scope:
430 if sparse_values.dtype == dtypes.string:
434 sparse_values, _ = sparse_ops.sparse_fill_empty_rows(
435 sparse_values, default_value)
437 segment_ids = sparse_values.indices[:, 0]
441 values = sparse_values.values
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_ops.py | 696 dense_values, dense_indices, sparse_values, sparse_indices = (
702 if sparse_values:
704 sparse_values)
743 def _do_batch_all_reduce_sparse(self, reduce_op, sparse_values): argument
748 len(sparse_values), 10)
752 reduce_op, zip(sparse_values, sparse_values))
|
| D | cross_device_utils.py | 706 sparse_values = []
710 sparse_values.append(v)
715 return dense_values, dense_indices, sparse_values, sparse_indices
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | sparse_to_dense_op_test.py | 32 sparse_values, argument
40 sparse_values,
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/ |
| D | input_data.cc | 106 const Tensor& sparse_values, in set_input_tensors() argument
115 new SparseValuesStorageType(sparse_values.tensor<float, 1>())); in set_input_tensors()
|
| D | input_data.h | 74 const Tensor& sparse_values,
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | sparse_ops_test.py | 106 def f(sparse_values, default_value): argument
109 values=sparse_values,
|
| D | parsing_ops.py | 869 (sparse_indices, sparse_values, sparse_shapes, dense_values) = outputs
873 in zip(sparse_indices, sparse_values, sparse_shapes)]
2122 (sparse_indices, sparse_values, sparse_shapes, dense_values) = outputs
2127 shape) in zip(sparse_indices, sparse_values, sparse_shapes)
|
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_fast_parsing.cc | 1103 result->sparse_values.emplace_back(config.sparse[d].dtype, values_shape); in FastParseExample()
1104 Tensor* values = &result->sparse_values.back(); in FastParseExample()
1289 result->sparse_values.emplace_back(); in FastParseSingleExample()
1496 out = &result->sparse_values[d]; in FastParseSingleExample()
1566 result->sparse_values[d] = in FastParseSingleExample()
1991 context_result->sparse_values.resize(context_config.sparse.size()); in FastParseSequenceExample()
1995 feature_list_result->sparse_values.resize(feature_list_config.sparse.size()); in FastParseSequenceExample()
2126 context_result->sparse_values[t] = Tensor(dtype, values_shape); in FastParseSequenceExample()
2135 out_bytes = context_result->sparse_values[t].flat<string>().data(); in FastParseSequenceExample()
2138 out_float = context_result->sparse_values[t].flat<float>().data(); in FastParseSequenceExample()
[all …]
|
| D | example_proto_fast_parsing.h | 88 std::vector<Tensor> sparse_values; member
|
| /external/tensorflow/tensorflow/contrib/linear_optimizer/python/ |
| D | sdca_optimizer.py | 128 sparse_values = array_ops.gather_nd(dense_tensor, sparse_indices)
138 array_ops.reshape(math_ops.cast(sparse_values, dtypes.float32), [-1]))
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| D | quantile_ops.cc | 422 const auto sparse_values = in Compute() local
437 stream.PushEntry(sparse_values(j), example_weights(example_id)); in Compute()
807 const auto sparse_values = in Compute() local
818 stream.PushEntry(sparse_values(j), example_weights(example_id)); in Compute()
|