Searched defs:result_shapes (Results 1 – 2 of 2) sorted by relevance
510 absl::Span<Shape const> result_shapes) { in CreateCallbackResults()543 absl::Span<Shape const> result_shapes, in MakePythonCallbackUsingHostSendAndRecv()592 absl::Span<Shape const> result_shapes) { in GetEmitPythonCallbackDescriptor()618 absl::Span<Shape const> result_shapes, in EmitPythonCallbackFromDescriptor()690 absl::Span<XlaOp const> operands, absl::Span<Shape const> result_shapes, in EmitPythonCallback()
327 std::vector<Shape> result_shapes; in MoveDimToMajor() local