| /external/tensorflow/tensorflow/core/kernels/ |
| D | searchsorted_op.cc | 87 const Tensor& values_t = ctx->input(1); in Compute() local
90 OP_REQUIRES(ctx, sorted_inputs_t.dim_size(0) == values_t.dim_size(0), in Compute()
95 OP_REQUIRES(ctx, values_t.NumElements() < std::numeric_limits<int>::max(), in Compute()
100 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, values_t.shape(), &output_t)); in Compute()
113 const auto values = values_t.template flat<T>(); in Compute()
117 sorted_inputs_t.dim_size(1), values_t.dim_size(1), &output)); in Compute()
128 const Tensor& values_t = ctx->input(1); in Compute() local
131 OP_REQUIRES(ctx, sorted_inputs_t.dim_size(0) == values_t.dim_size(0), in Compute()
136 OP_REQUIRES(ctx, values_t.NumElements() < std::numeric_limits<int>::max(), in Compute()
141 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, values_t.shape(), &output_t)); in Compute()
[all …]
|
| D | sparse_softmax_op.cc | 43 const Tensor *indices_t, *values_t, *shape_t; in Compute() local
45 OP_REQUIRES_OK(context, context->input("sp_values", &values_t)); in Compute()
54 TensorShapeUtils::IsVector(values_t->shape()) && in Compute()
59 values_t->shape().DebugString(), " and ", in Compute()
71 tensor::DeepCopy(*indices_t), tensor::DeepCopy(*values_t), in Compute()
|
| D | sparse_dense_binary_op_shared.cc | 62 const Tensor *indices_t, *values_t, *shape_t, *dense_t; in Compute() local
64 OP_REQUIRES_OK(ctx, ctx->input("sp_values", &values_t)); in Compute()
74 TensorShapeUtils::IsVector(values_t->shape()) && in Compute()
79 values_t->shape().DebugString(), " and ", in Compute()
160 values_t->flat<T>().binaryExpr(dense_gathered_flat, in Compute()
|
| D | sparse_fill_empty_rows_op.cc | 43 const Tensor& indices_t, const Tensor& values_t, in operator ()()
52 const auto values = values_t.vec<T>(); in operator ()()
131 context->set_output(kOutputValuesOutput, values_t); in operator ()()
201 const Tensor& values_t = context->input(kValuesInput); in Compute() local
211 OP_REQUIRES(context, TensorShapeUtils::IsVector(values_t.shape()), in Compute()
213 values_t.shape().DebugString())); in Compute()
221 context, default_value_t, indices_t, values_t, in Compute()
|
| D | sparse_reduce_op.cc | 161 const Tensor *indices_t, *values_t, *shape_t, *reduction_axes_t; in Compute() local
163 OP_REQUIRES_OK(ctx, ctx->input("input_values", &values_t)); in Compute()
177 tensor::DeepCopy(*indices_t), tensor::DeepCopy(*values_t), in Compute()
257 const Tensor *indices_t, *values_t, *shape_t, *reduction_axes_t; in Compute() local
259 OP_REQUIRES_OK(ctx, ctx->input("input_values", &values_t)); in Compute()
267 tensor::DeepCopy(*values_t), in Compute()
|
| D | ctc_decoder_ops.cc | 147 auto values_t = p_values->vec<int64>(); in StoreAllDecodedSequences() local
158 DCHECK_NE(values_t.data(), nullptr) in StoreAllDecodedSequences()
161 DCHECK_LT(offset, values_t.size()) in StoreAllDecodedSequences()
163 std::copy_n(p_batch.begin(), num_decoded, &values_t(offset)); in StoreAllDecodedSequences()
|
| D | sparse_fill_empty_rows_op.h | 30 const Tensor& indices_t, const Tensor& values_t,
|
| D | reader_ops.cc | 146 auto values_t = values->vec<tstring>(); in ComputeWithReader() local
149 values_t(i) = std::move(values_vec[i]); in ComputeWithReader()
|
| D | ragged_cross_op.cc | 126 SparseFeatureReader(const Tensor& indices_t, const Tensor& values_t, in SparseFeatureReader() argument
128 : values_(values_t.flat<ValuesType>()) { in SparseFeatureReader()
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | dense_to_csr_sparse_matrix_op.cc | 241 const Tensor& values_t = const_cast<const Tensor&>(temp_values_t); in ComputeAsync() local
244 c, TensorShapeUtils::IsVector(values_t.shape()), in ComputeAsync()
246 values_t.shape().DebugString()), in ComputeAsync()
267 c, total_nnz == values_t.NumElements(), in ComputeAsync()
271 total_nnz, " vs. ", values_t.NumElements()), in ComputeAsync()
276 Tensor csr_values_t = values_t; in ComputeAsync()
333 values_t.dtype(), dense_shape_t, batch_ptr_t, csr_row_ptr_t, in ComputeAsync()
|
| D | sparse_tensor_to_csr_sparse_matrix_op.cc | 126 const Tensor& values_t = c->input(1); in ComputeAsync() local
176 nnz_per_batch_device_ref, stream, &d, &values_t, in ComputeAsync()
200 c, total_nnz == values_t.NumElements(), in ComputeAsync()
204 total_nnz, " vs. ", values_t.NumElements()), in ComputeAsync()
209 Tensor csr_values_t = values_t; in ComputeAsync()
266 values_t.dtype(), dense_shape_t, batch_ptr_t, csr_row_ptr_t, in ComputeAsync()
|
| D | sparse_matrix_components_op.cc | 76 Tensor* values_t; in Compute() local
80 OP_REQUIRES_OK(c, c->allocate_output(2, TensorShape({nnz}), &values_t)); in Compute()
83 auto values = values_t->vec<T>(); in Compute()
|
| D | csr_sparse_matrix_to_dense_op.cc | 154 Tensor values_t; in Compute() local
156 TensorShape({total_nnz}), &values_t)); in Compute()
191 values_t = csr_sparse_matrix->values(); in Compute()
201 c, indices_t, values_t, dense_tensor_shape, &dense_t, in Compute()
|
| D | csr_sparse_matrix_to_sparse_tensor_op.cc | 160 Tensor* values_t; in Compute() local
162 c->allocate_output(1, TensorShape({total_nnz}), &values_t)); in Compute()
197 *values_t = csr_sparse_matrix->values(); in Compute()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | sparse_ops_test.py | 99 values_t = constant_op.constant(values)
101 indices=indices_t, values=values_t, dense_shape=dense_shape_t)
|
| /external/rust/crates/clap/src/ |
| D | macros.rs | 157 macro_rules! values_t { macro
159 values_t!($m.values_of($v), $t)
|