Searched defs:return_shape (Results 1 – 9 of 9) sorted by relevance
128 const Shape& return_shape = root_instruction->shape(); in CodegenNestedComputation() local
633 const Shape& return_shape = computation.root_instruction()->shape(); in ComputeNestedElement() local
950 const Shape* return_shape = &shape; in GetSubshape() local961 const Shape* return_shape = &shape; in TryGetSubshape() local976 Shape* return_shape = shape; in GetMutableSubshape() local
899 Shape return_shape; in TransferToClient() local
121 const Shape& return_shape = computation->root_instruction()->shape(); in EmitThreadLocalFunctionEpilogue() local3273 const Shape& return_shape = callee.root_instruction()->shape(); in EmitThreadLocalCall() local
1793 xla::Shape return_shape = xla::ShapeUtil::MakeTupleShape( in TEST() local1892 xla::Shape return_shape = xla::ShapeUtil::MakeTupleShape( in TEST() local
1514 xla::Shape return_shape = xla::TypeToShape(ret.get().getType()); in Lower() local
1656 SmallVector<int64_t, 4> return_shape(num_dims, ShapedType::kDynamicSize); in inferConvReturnTypes() local
1107 SmallVector<int64_t, 4> return_shape(4, ShapedType::kDynamicSize); in inferReturnTypes() local