/external/tensorflow/tensorflow/core/kernels/ |
sparse_fill_empty_rows_op.cc | 97 Tensor* output_indices_t; variable 101 &output_indices_t)); 139 Tensor* output_indices_t; variable 144 &output_indices_t)); 145 auto output_indices = output_indices_t->matrix<int64>();
|
sparse_sparse_binary_op_shared.cc | 194 Tensor *output_indices_t, *output_values_t; variable 197 &output_indices_t)); 200 auto output_indices_mat = output_indices_t->matrix<int64>();
|
serialize_sparse_op.cc | 222 auto output_indices_t = output_indices.matrix<int64>(); variable 227 output_indices_t(i, d - 1) = indices(i, d); 367 const auto& output_indices_t = output_indices.matrix<int64>(); variable 372 expanded_indices_t.slice(indices_start, indices_sizes) = output_indices_t;
|
sparse_tensors_map_ops.cc | 293 auto output_indices_t = output_indices.matrix<int64>(); variable 298 output_indices_t(i, d - 1) = indices(i, d); 419 const auto& output_indices_t = output_indices.matrix<int64>(); variable 425 expanded_indices_t.slice(indices_start, indices_sizes) = output_indices_t;
|
/external/tensorflow/tensorflow/core/util/sparse/ |
sparse_tensor.h | 505 std::vector<typename TTypes<int64>::Matrix> output_indices_t; local 507 output_indices_t.reserve(num_split); 538 output_indices_t.emplace_back(output_indices[i].matrix<int64>()); 552 output_indices_t[slice_index](slice_dim, j) = 613 auto output_indices_t = output_indices.matrix<int64>(); 635 output_indices_t(index, dim) = input_indices_t(i, dim) - start[dim];
|