Searched refs:out_values_t (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_add_op.cc | 147 Tensor *out_indices_t, *out_values_t; in Compute() local
152 ctx, ctx->allocate_output(1, TensorShape({sum_nnz}), &out_values_t)); in Compute()
154 auto out_values_flat = out_values_t->vec<T>(); in Compute()
|
| D | sparse_reduce_op.cc | 290 Tensor *out_values_t; in Compute() local
292 ctx->allocate_output(1, TensorShape({nnz}), &out_values_t)); in Compute()
293 auto out_flat = out_values_t->flat<T>(); in Compute()
|
| D | set_kernels.cc | 146 Tensor *out_indices_t, *out_values_t, *out_shape_t; in OutputSparseTensor() local
151 ctx, ctx->allocate_output(1, TensorShape({num_values}), &out_values_t)); in OutputSparseTensor()
155 auto out_values_flat = out_values_t->vec<T>(); in OutputSparseTensor()
|