- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for GenArgShapeInfos (0.12 sec)
-
tensorflow/compiler/aot/codegen.cc
*methods += RewriteWithName( var.name().empty() ? var.node_name() : var.name(), code, rewrites); } return absl::OkStatus(); } // Generate shape infos for args (inputs). Status GenArgShapeInfos(const xla::ProgramShapeProto& ps, string* infos) { for (int i = 0; i < ps.parameters_size(); ++i) { const xla::ShapeProto& shape = ps.parameters(i); if (shape.element_type() == xla::TUPLE) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 01:20:01 UTC 2024 - 36.8K bytes - Viewed (0)