Searched defs:return_shape (Results 1 – 9 of 9) sorted by relevance
128 const Shape& return_shape = root_instruction->shape(); in CodegenNestedComputation() local
638 const Shape& return_shape = computation.root_instruction()->shape(); in ComputeNestedElement() local
910 const Shape* return_shape = &shape; in GetSubshape() local921 const Shape* return_shape = &shape; in TryGetSubshape() local936 Shape* return_shape = shape; in GetMutableSubshape() local
921 Shape return_shape; in TransferToClient() local
121 const Shape& return_shape = computation->root_instruction()->shape(); in EmitThreadLocalFunctionEpilogue() local3236 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
1319 xla::Shape return_shape = xla::TypeToShape(ret.get().getType()); in Lower() local
1633 SmallVector<int64_t, 4> return_shape(num_dims, ShapedType::kDynamicSize); in inferConvReturnTypes() local
1320 SmallVector<int64_t, 4> return_shape(input_rank, ShapedType::kDynamicSize); in inferReturnTypes() local