Searched refs:shape_getter (Results 1 – 2 of 2) sorted by relevance
81 const std::function<xla::Shape(int64)>& shape_getter, bool release_inputs);
259 const std::function<xla::Shape(int64)>& shape_getter, bool release_inputs) { in GetInputTupleAllocations() argument278 xla::Shape input_shape = shape_getter(i); in GetInputTupleAllocations()