Searched defs:result_shapes (Results 1 – 3 of 3) 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()
92 std::vector<Shape>& result_shapes) { in applySymbolicShapesOnLT()
327 std::vector<Shape> result_shapes; in MoveDimToMajor() local