Searched defs:result_shapes (Results 1 – 3 of 3) sorted by relevance
510 absl::Span<XlaOp const> operands, absl::Span<Shape const> result_shapes, in EmitPythonCallback()
128 SmallVector<Value> result_shapes; in matchAndRewrite() local
324 std::vector<Shape> result_shapes; in MoveDimToMajor() local