Searched refs:output_values_t (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_fill_empty_rows_op.cc | 87 Tensor* output_values_t; in operator ()() local
89 kOutputValuesOutput, TensorShape({0}), &output_values_t)); in operator ()()
145 Tensor* output_values_t; in operator ()() local
147 kOutputValuesOutput, TensorShape({N_full}), &output_values_t)); in operator ()()
148 auto output_values = output_values_t->vec<T>(); in operator ()()
|
| D | sparse_sparse_binary_op_shared.cc | 194 Tensor *output_indices_t, *output_values_t; in Compute() local
199 ctx, ctx->allocate_output(1, TensorShape({sum_nnz}), &output_values_t)); in Compute()
223 output_values_t->flat<T>().device(ctx->eigen_device<Device>()) = in Compute()
|
| D | serialize_sparse_op.cc | 197 auto output_values_t = output_values.vec<T>(); in operator ()() local
203 output_values_t(i) = values(i); in operator ()()
|
| D | sparse_tensors_map_ops.cc | 300 auto output_values_t = output_values.vec<T>(); in Compute() local
306 output_values_t(i) = values(i); in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | transpose_op.cc | 172 Tensor output_values_t; in operator ()() local
180 DataTypeToEnum<T>::value, TensorShape({total_nnz}), &output_values_t)); in operator ()()
184 output_row_ptr_t, output_col_ind_t, output_values_t, output_matrix)); in operator ()()
208 maybe_conj_inplace<Device, T>::run(d, &output_values_t); in operator ()()
|
| D | softmax_op.cc | 59 Tensor output_values_t; in Compute() local
62 TensorShape({total_nnz}), &output_values_t)); in Compute()
73 logits_matrix->col_indices(), output_values_t, &output_matrix)); in Compute()
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | sparse_tensor.h | 506 std::vector<typename TTypes<T>::Vec> output_values_t; in Split() local
508 output_values_t.reserve(num_split); in Split()
541 output_values_t.emplace_back(output_values[i].vec<T>()); in Split()
551 output_values_t[slice_index](slice_dim) = input_values_t(i); in Split()
631 auto output_values_t = output_values.vec<T>(); in Slice() local
652 output_values_t(index) = input_values_t(i); in Slice()
|