Searched defs:argument_layouts (Results 1 – 13 of 13) sorted by relevance
101 const absl::Span<const Shape* const> argument_layouts, in GetHloModuleConfig()154 const absl::Span<const Shape* const> argument_layouts, in CompileExecutables()201 const absl::Span<const Shape* const> argument_layouts, in CompileAotResults()
44 std::vector<const Shape*> argument_layouts; member
389 const absl::Span<const Shape* const> argument_layouts, in Compile()412 const absl::Span<const Shape* const> argument_layouts, in CompileAheadOfTime()
49 std::vector<const xla::Shape*> argument_layouts( in BuildExecutable() local
284 std::vector<const xla::Shape*> argument_layouts = in BuildExecutable() local302 std::vector<const xla::Shape*> argument_layouts = in BuildSerializedExecutable() local
153 std::vector<xla::Shape> argument_layouts( in Compile() local
152 std::optional<std::vector<Shape>>& argument_layouts, in DetermineArgumentLayoutsFromCompileOptions()
220 std::optional<std::vector<Shape>> argument_layouts; member
248 const absl::Span<const Shape* const> argument_layouts, in JitCompile()
189 std::vector<const Shape*> argument_layouts(arguments.size()); in ExecuteLocally() local
806 std::optional<std::vector<Shape>> argument_layouts, in Compile()879 std::optional<std::vector<Shape>> argument_layouts, in CompileMlir()
116 std::vector<Shape> argument_layouts; in CompileExecutable() local