HomeSort by relevance Sort by last modified time
    Searched full:pshape (Results 1 - 1 of 1) sorted by null

  /external/tensorflow/tensorflow/compiler/aot/
compile.cc 58 xla::ProgramShape* pshape = &compile_result->program_shape; local
60 arg_layouts.reserve(pshape->parameters_size());
61 for (int i = 0; i < pshape->parameters_size(); ++i) {
62 arg_layouts.push_back(pshape->mutable_parameters(i));
67 instance.result_layout = &pshape->result();

Completed in 293 milliseconds