Searched defs:return_shape (Results 1 – 6 of 6) sorted by relevance
120 const Shape& return_shape = root_instruction->shape(); in CodegenNestedComputation() local
778 const Shape& return_shape = function->root_instruction()->shape(); in HandleReduce() local
783 const Shape* return_shape = &shape; in GetSubshape() local794 const Shape* return_shape = &shape; in TryGetSubshape() local809 Shape* return_shape = shape; in GetMutableSubshape() local
965 Shape return_shape; in TransferToClient() local
118 const Shape& return_shape = computation->root_instruction()->shape(); in EmitThreadLocalFunctionEpilogue() local3363 const Shape& return_shape = callee.root_instruction()->shape(); in EmitThreadLocalCall() local
1385 xla::Shape return_shape = xla::ShapeUtil::MakeTupleShape( in TEST() local