- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for kArgShapeInfoTable (0.11 sec)
-
tensorflow/compiler/aot/codegen_test_h.golden
static constexpr int32_t kArg4Shapes[] = { 5 }; static const ShapeInfo* ArgShapeInfos() { static constexpr ShapeInfo kArgShapeInfoTable[kNumArgs] = { { kArg0Shapes, 2 }, { kArg1Shapes, 2 }, { kArg2Shapes, 1 }, { kArg3Shapes, 1 }, { kArg4Shapes, 1 }, }; return kArgShapeInfoTable; }; // Shapes of the results. static constexpr int32_t kResult0Shapes[] = { 5, 6 };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 01:20:01 UTC 2024 - 16.6K bytes - Viewed (0) -
tensorflow/compiler/aot/codegen.cc
static constexpr ShapeInfo kArgShapeInfoTable[kNumArgs] = { )"; for (int i = 0; i < ps.parameters_size(); ++i) { const xla::ShapeProto& shape = ps.parameters(i); *infos += absl::Substitute("{ kArg$0Shapes, $1 },\n", i, shape.dimensions_size()); } *infos += R"( }; return kArgShapeInfoTable; })"; return absl::OkStatus(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 01:20:01 UTC 2024 - 36.8K bytes - Viewed (0)