Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dtranspose.h172 int64_t num_elems_; variable
Dtranspose.cc457 if (num_elems_ == 0) { in Execute()
920 plan->num_elems_ = std::accumulate(dims.begin(), dims.end(), int64_t{1}, in Create()
1029 if (num_elems_ == 0) { in Initialize()