Home
last modified time | relevance | path

Searched refs:out_shape_flat (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsparse_reduce_op.cc321 auto out_shape_flat = out_shape_t->flat<int64>(); in Compute() local
323 std::copy(out_dim_sizes.begin(), out_dim_sizes.end(), &out_shape_flat(0)); in Compute()
Dset_kernels.cc183 auto out_shape_flat = out_shape_t->vec<int64>(); in OutputSparseTensor() local
185 out_shape_flat(i) = output_shape.dim_size(i); in OutputSparseTensor()