Searched defs:argument_layouts (Results 1 – 10 of 10) sorted by relevance
44 std::vector<const Shape*> argument_layouts; member
342 const absl::Span<const Shape* const> argument_layouts, in Compile()
125 const absl::Span<const Shape* const> argument_layouts, in CompileExecutables()
113 std::vector<xla::Shape> argument_layouts( in Compile() local
192 std::vector<const Shape*> argument_layouts(arguments.size()); in ExecuteLocally() local
723 absl::optional<std::vector<Shape>> argument_layouts, in CompileForDevices()769 absl::optional<std::vector<Shape>> argument_layouts, in Compile()
153 std::vector<const xla::Shape*> argument_layouts( in BuildExecutable() local
966 absl::optional<std::vector<Shape>> argument_layouts, in CompileForDevices()1012 absl::optional<std::vector<Shape>> argument_layouts, in Compile()
117 std::vector<Shape> argument_layouts; in CompileExecutable() local