Searched defs:argument_layouts (Results 1 – 11 of 11) sorted by relevance
44 std::vector<const Shape*> argument_layouts; member
358 const absl::Span<const Shape* const> argument_layouts, in Compile()
103 const absl::Span<const Shape* const> argument_layouts, in CompileExecutables()
117 absl::optional<std::vector<Shape>> argument_layouts; member
150 absl::optional<std::vector<Shape>>& argument_layouts, in DetermineArgumentLayoutsFromCompileOptions()
152 std::vector<xla::Shape> argument_layouts( in Compile() local
192 std::vector<const Shape*> argument_layouts(arguments.size()); in ExecuteLocally() local
163 std::vector<const xla::Shape*> argument_layouts( in BuildExecutable() local
117 std::vector<Shape> argument_layouts; in CompileExecutable() local
796 absl::optional<std::vector<Shape>> argument_layouts, in Compile()