HomeSort by relevance Sort by last modified time
    Searched refs:out_shape_flat (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
sparse_reduce_op.cc 317 auto out_shape_flat = out_shape_t->flat<int64>(); variable
319 std::copy(out_dim_sizes.begin(), out_dim_sizes.end(), &out_shape_flat(0));
set_kernels.cc 188 auto out_shape_flat = out_shape_t->vec<int64>(); local
190 out_shape_flat(i) = output_shape.dim_size(i);

Completed in 259 milliseconds